203_remove_linked_list_elements

203. 移除链表元素 简单

给你一个链表的头节点 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 <= k <= 50

代码参考：

package main

import "fmt"

func main() {
    fmt.Println(removeElements(newList([]int{1, 1}), 1))
    fmt.Println(removeElements(newList([]int{1, 2, 6, 3, 4, 5, 6}), 6))
}

// 头节点可能会被删掉，要考虑借助哑节点
// 画图理解就 ok
func removeElements(head *ListNode, val int) *ListNode {
    if head == nil {
        return head
    }
    dummy := &ListNode{Next: head}
    pre, cur := dummy, head
    for cur != nil {
        if cur.Val == val {
            pre.Next = cur.Next
            cur = cur.Next
            continue
        }
        pre = pre.Next
        cur = cur.Next
    }
    return dummy.Next
}