LeetCode刷题之路: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

#最直观的思路

首先在头部新建一个虚拟节点,
并将该节点指向head
此做法是为了防止需要删除首节点的情况。
随后开始判断,如果当前节点的下一个节点的val需要删除
则直接将当前节点的下一节点换为下一节点的下一节点
若不需要删除,则直接将当前节点换为下一节点即可
直至所有节点遍历完毕

最后需要注意的是 我们需要返回虚拟节点的下一个位置才可以

class Solution:
    def removeElements(self, head: ListNode, val: int) -> ListNode:
        now = ListNode(0, head)
        l1 = now
        # print(l1)
        while l1.next != None:
            if l1.next.val == val:
                l1.next = l1.next.next
            else:
                l1 = l1.next
        return now.next

猜你喜欢

转载自blog.csdn.net/as812252319/article/details/115275292