链表篇 203. 移除链表元素

  1. 移除链表元素 (难度 简单) 给你一个链表的头节点 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

题解: 两种方法: 因为链表的其他节点都是通过前一个节点来移除当前节点,而头结点没有前一个节点。 方法一:设置一个虚拟头结点再进行删除操作(正所谓没有条件创造条件)

class Solution:
    def removeElements(self, head: ListNode, val: int) -> ListNode:
        dummy_head = ListNode(next=head) #虚拟头节点
        cur = dummy_head 
        while cur.next!=None:
            if cur.next.val==val:
                cur.next=cur.next.next 
            else:
                cur=cur.next
        return dummy_head.next

方法二:还是用原来的链表来进行删除操作。

class Solution:
    def removeElements(self, head: ListNode, val: int) -> ListNode:
        while head and head.val == val: #若头结点值为val,head就指向下一个
            head = head.next
        pre,cur = head,head #pre 进行删除 cur负责判断移动
        while cur: 
            if cur.val == val:
                pre.next = cur.next
            else:
                pre = cur
            cur = cur.next
        return head

推荐方法一 简单易懂

时间复杂度:O(n),其中 nn是链表的长度。需要遍历链表一次。 空间复杂度:O(1)。

经验分享 程序员 微信小程序 职场和发展