python移除链表元素

最新推荐文章于 2024-05-29 12:15:53 发布

原创最新推荐文章于 2024-05-29 12:15:53 发布 · 909 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#链表 #python #数据结构

算法同时被 2 个专栏收录

78 篇文章

订阅专栏

链表

16 篇文章

订阅专栏

该博客介绍了一个Python实现的解决方案，用于从链表中删除所有值等于给定整数的节点。通过创建虚拟头节点，处理头节点、尾节点及空链表的特殊情况，实现了高效地遍历并删除目标节点。最终返回新的链表头节点。

｜移除链表元素

给你一个链表的头节点 head 和一个整数 val ，请你删除链表中所有满足 Node.val == val 的节点，并返回新的头节点。
在这里插入图片描述
输入：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
输出：[]

｜题解

# 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: ListNode, val: int) -> ListNode:
        # 解决链表常见的三种情况
        # 1.头节点是否需要特殊处理
        # 2.尾节点是否需要特殊处理
        # 3.空链表是否需要特殊处理
        
        # 常见的处理方式 引入虚拟头节点
        # 引入虚拟头节点
        new_head = ListNode
        new_head.next = head
        prev = new_head
        while prev.next is not None:
            # 如果要删除某个节点，首先要找到被删除节点的前驱节点
            if prev.next.val == val:
                prev.next = prev.next.next
            else:
                prev = prev.next
        return new_head.next