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 <= val <= 50
单链表,构建虚拟头结点,因为Golang已经提供了GC机制,所以无需手动删除。
需要注意的就是不能直接拿head遍历,不然最后找不到head的位置了。
go
type ListNode struct {
Val int
Next *ListNode
}
// 构建虚拟头结点
func removeElements(head *ListNode, val int) *ListNode {
dummyhead := &ListNode{0, head}
cur := dummyhead
for cur.Next != nil {
if cur.Next.Val == val {
cur.Next = cur.Next.Next
} else {
cur = cur.Next
}
}
return dummyhead.Next
}