1171. 从链表中删去总和值为零的连续节点

第一次遍历,记录所有前缀和的最后出现指针。

然后,我们能证明:将中间相加为0的节点删除后,其它节点的前缀和都不变。比如对于[1, 3, -3, 4],删除[3,-3]以后,1和4所在位置的前缀和还是不会变,这是因为中间数字的和为零。

  • 1的前缀和不会变,因为删除的位置在此之后
  • 4的前缀和不会变,因为在此前删除掉的节点和为零。

第二次遍历时,缓存m里为当前的前缀和存的节点,只可能在当前或之后出现。

  • 它不可能在此之前,因为如果有,那m里存的就不是最晚的节点了,与设定矛盾
  • 可以就是当前,或者之后。如果在之后,将这一段和为零的节点都删去,然后看向之后的节点。删除操作不会更改之后节点的前缀和。
class Solution:
    def removeZeroSumSublists(self, head: ListNode) -> ListNode:
        m = dict()
        foo = ListNode(0)

        foo.next = head
        cur = foo
        total = 0
        
        while cur:
            total += cur.val
            m[total] = cur
            cur = cur.next
        
        cur = foo
        total = 0
        while cur:
            total += cur.val
            cur.next = m[total].next
            cur = cur.next
        return foo.next
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值