1171 从链表中删去总和值为零的连续节点(前缀和、分析)

1. 问题描述:

给你一个链表的头节点 head,请你编写代码,反复删去链表中由总和值为0 的连续节点组成的序列,直到不存在这样的序列为止。删除完毕后,请你返回最终结果链表的头节点。你可以返回任何满足题目要求的答案。(注意,下面示例中的所有序列,都是对 ListNode 对象序列化的表示)

示例 1:
输入:head = [1,2,-3,3,1]
输出:[3,1]
提示:答案 [1,2,1] 也是正确的。

示例 2:
输入:head = [1,2,3,-3,4]
输出:[1,2,4]

示例 3:
输入:head = [1,2,3,-3,-2]
输出:[1]

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/remove-zero-sum-consecutive-nodes-from-linked-list

2. 思路分析:

① 一开始的时候没有啥思路,于是看了一下领扣中关于python的解法,其中发现一个很不错的思路,我结合了题解的代码使用测试用例理解了一下,下面是我对于题解代码思路的一些理解,主要的思路如下:其中比较关键是利用一个累加到当前位置的总和如果出现一个与之前一样的结果那么说明中间肯定是存在连续的总和为0的序列,这个时候我们只需要将这个连续总和为0的序列的前面与后面连接起来即可,这里可以可以借助于python中的字典来做辅助的作用,因为使用字典的时候当结果相同的时候回保存最新的一个节点的值

② 这里需要借助于一个虚拟的一个值为0的节点做辅助可以避免链表为空的情况,①中字典可以保存对应节点的值,当出现相同键的时候这个时候保存的是最新节点的值,这样就可以使中间总和为0的连续的序列跳过去了,所以我们最后使用一个for循环来连接这些节点即可,感觉这个思路还是非常巧妙的,可以结合pycharm中的debug测试一下理解一下其中巧妙的思路

3. 代码如下:

class ListNode:
    def __init__(self, x):
        self.val = x
        self.next = None

class Solution:
    def removeZeroSumSublists(self, head: ListNode) -> ListNode:
        seen = dict()
        profix = 0
        dummy = ListNode(0)
        dummy.next = head
        seen[0] = dummy
        while head:
            profix += head.val
            seen[profix] = head
            head = head.next

        head = dummy
        profix = 0

        while head:
            profix += head.val
            head.next = seen[profix].next
            head = head.next
        return dummy.next


if __name__ == '__main__':
    head = ListNode(1)
    n1 = ListNode(2)
    next1 = ListNode(2)
    nnext = ListNode(-3)
    nnnext = ListNode(3)
    nnnnext = ListNode(1)
    head.next = n1
    n1.next = next1
    next1.next = nnext
    nnext.next = nnnext
    nnnext.next = nnnnext
    nnnnext.next = None
    res = Solution().removeZeroSumSublists(head)
    while res:
        print(res.val)
        res = res.next

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值