您现在的位置是:首页 >技术杂谈 >【链表】力扣203题:移除链表元素网站首页技术杂谈

【链表】力扣203题:移除链表元素

三耳01 2023-06-07 11:30:35
简介【链表】力扣203题:移除链表元素

【链表】力扣203题:移除链表元素

建议在看题目之前先了解数组的具体知识点,可以看这里:
算法基础(三):链表知识点及题型讲解

力扣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
输出:[]

来源:力扣(LeetCode)203

把所有等于val的节点移除掉:

  • 对示例1head = [1,2,6,3,4,5,6], val = 6,引入一个临时节点(dummy节点),放在头节点1的前面,则dummy.next就是head = [1,2,6,3,4,5,6]这一串。
  • head移动到了第一个6的时候,要把6删除,那就是把2next指针指向3,此时,3好说,就是head.next,但是2不好说,所以引入一个prev指针,代表head前面的位置,即:在每次head移动之前,先将head对应的值赋值给prev,再移动head.next
    那么,当删除6的时候,就是直接prev.next = head.next
# Definition for singly-linked list.
# class ListNode:
#     def __init__(self, val=0, next=None):
#         self.val = val
#         self.next = next
class Solution:
    def removeElements(self, head: Optional[ListNode], val: int) -> Optional[ListNode]:
        # head是第一个节点的位置
        dummy = ListNode(0)  # 括号里面随便一个值都行,这个数用不到
        dummy.next = head  # 用dummy.next表示head,便于后续返回
        prev = dummy  # prev是head前面的指针
        while head is not None:
            if head.val == val:
                prev.next = head.next
            else:
                prev = head
            head = head.next
        return dummy.next

执行用时:72 ms, 在所有 Python3 提交中击败了23.53%的用户
内存消耗:18.4 MB, 在所有 Python3 提交中击败了35.57%的用户

风语者!平时喜欢研究各种技术,目前在从事后端开发工作,热爱生活、热爱工作。