自学内容网 自学内容网

Leetcode203.移除链表元素-Python

题目链接:203. 移除链表元素 - 力扣(LeetCode)

题目描述:

给你一个链表的头节点 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
输出:[]

解题思路:虚拟头节点

考虑到如果要移除的元素如果是头节点的话需要head = head.next,那么每次都需要判断,头节点和非头节点需要不同的处理方式较为复杂,所以此处引入虚拟头节点避免了这一问题。

移除元素时需要知道上一个元素,将上一个元素指向要移除元素的下一个元素,所以需要初始化一个指针,指向当前节点,遍历链表进行循环,循环条件是当前指针的下一个元素不为空。

代码:

class ListNode:
    def __init__(self,val = 0,next = None):
        self.val = val
        self.next = next
class Solution:
    def removeElements(self,head:Optional[ListNode],target:int)->Optional[ListNode]:
        # 初始化虚拟头节点
        dummy_head = ListNode(next = head)
        # 初始化指针
        cur = dummy_head
        while cur.next:
              if cur.next.val == target
                 cur.next = cur.next.next
              else:
                 cur = cur.next
        return dummy_head.next
        


原文地址:https://blog.csdn.net/qq_64631161/article/details/142741745

免责声明:本站文章内容转载自网络资源,如本站内容侵犯了原著者的合法权益,可联系本站删除。更多内容请关注自学内容网(zxcms.com)!