Leetcode 203 移除链表元素

最新推荐文章于 2024-09-07 19:07:25 发布

苹果就叫不呐呐

最新推荐文章于 2024-09-07 19:07:25 发布

阅读量90

点赞数

分类专栏： Leetcode # 链表文章标签：链表 leetcode 数据结构

本文链接：https://blog.csdn.net/prince0520/article/details/129773156

版权

Leetcode 同时被 2 个专栏收录

93 篇文章 0 订阅

订阅专栏

链表

9 篇文章 0 订阅

订阅专栏

本文介绍了LeetCode第203题的两种解决方案，均具有O(N)的时间复杂度和O(1)的空间复杂度。解法一通过在头节点前插入虚拟节点来处理删除，解法二同样利用虚拟节点，但在遍历过程中直接更新头节点。

摘要由CSDN通过智能技术生成

Leetcode 203 移除链表元素

- 解法1 O(N) O(1)
- 解法2 O(N) O(1)

解法1 O(N) O(1)


class Solution:
    def removeElements(self, head: Optional[ListNode], val: int) -> Optional[ListNode]:
        # head头结点 
        #self.val = val
        #self.next = next
        # 需要对于头结点单独操作（如果头结点=val的话 就没办法断开指针重新连接）
        
        while head is not None:
            if head.val == val:
                head = head.next # 直接右移一个head
            else:
                break

         # 在原来链表前面引入一个虚拟节点dummy
         # 引入一个prev头 在head前，删除元素时：令prev.next=head.next         
        dummy = ListNode(-1)
        dummy.next = head
        prev = dummy
        now = head
        while now is not None:  
            if now.val == val:
                prev.next= now.next
            else:
                prev = prev.next
            now = now.next
        return head # 返回新的头结点

在这里插入图片描述

解法2 O(N) O(1)

        # 引入一个prev头 在head前，删除元素时：令prev.next=head.next
        # 在原来链表前面引入一个虚拟节点dummy
        dummy = ListNode(-1)
        dummy.next = head
        prev = dummy
        # 之后不断遍历head 最后返回的是dummy.next，相当于一开始存下来了head
        while head is not None:  
            if head.val == val: #删除元素
                prev.next= head.next
            else:#继续遍历
                prev = prev.next 
            head = head.next # head头继续移动

        return dummy.next # 返回