链表篇 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
题解: 两种方法: 因为链表的其他节点都是通过前一个节点来移除当前节点,而头结点没有前一个节点。 方法一:设置一个虚拟头结点再进行删除操作(正所谓没有条件创造条件)
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)。
下一篇:
[烂莓子的c++学习笔记](二)函数重载