题目描述:
删除链表中等于给定值 val 的所有节点。
示例:
输入: 1->2->6->3->4->5->6, val = 6 输出: 1->2->3->4->5
链表元素的操作没什么好纠结的,上指针就行,一个不够就上两个,注意判断非空以及终止。
(1)定义链表结构:
class ListNode:
def __init__(self, x):
self.val = x
self.next = None
(2)初始化头节点
init = ListNode(0)
(2)顺序查找和删除
init.next = head
pre,cur = init,head
while cur:
if cur.val == val:
cur = cur.next
pre.next = cur
else:
pre = cur
cur = cur.next
return init.next
解释一下:
初始化一个头节点 init
然后定义2个指针:cur、pre 指向当前节点和当前节点的上一节点
当前节点非空时,判断是否是要删除的值val,
节点的删除通过调整节点指针实现,cur = cur.next pre.next = cur
如果不需要删除:pre = cur cur = cur.next
(3)完整代码
class Solution:
def removeElements(self, head: ListNode, val: int) -> ListNode:
newH = ListNode(0)
newH.next = head
pre, p = newH, head
while p:
if p.val == val:
pre.next = p = p.next
else:
pre, p = p, p.next
return newH.next