第一次遍历,记录所有前缀和的最后出现指针。
然后,我们能证明:将中间相加为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