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

你可能感兴趣的:(数据结构与算法,链表,数据结构,leetcode)