203. 移除链表元素

leecode原题

题目

给你一个链表的头节点 head 和一个整数 val ，请你删除链表中所有满足 Node.val == val 的节点，并返回 新的头节点 。

示例

示例 1:

输入：head = [1,2,6,3,4,5,6], val = 6
输出：[1,2,3,4,5]

示例 2:

输入：head = [], val = 1
输出：[]

示例 3:

输入：head = [7,7,7,7], val = 7
输出：[]

提示：

• 列表中的节点数目在范围 [0, 104] 内
• 1 <= Node.val <= 50
• 0 <= val <= 50

解题思路

思路

这个题本身就是考察对链表进行删除的脚本操作, 唯一的一个关注点就是，因为考虑到比如值刚好等于头节点本身这种情况，正常我们可能需要去判断相关的情况做处理，但是我们增加一个在头结点之前增加一个虚拟头结点，这样头结点就跟其它节点处理是一样的了，最后返回的时候，我们返回虚拟头结点的next节点即可。

实现

源码

func removeElements(head *ListNode, val int) *ListNode {
    virtualHead := &ListNode{}
    virtualHead.Next = head
    cur_node := virtualHead
    for cur_node != nil && cur_node.Next != nil {
        if cur_node.Next.Val == val {
            cur_node.Next = cur_node.Next.Next
        } else {
            cur_node = cur_node.Next
        }
    }
    return virtualHead.Next
}