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

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

把链表存进数组，用字典记录前缀和所在的坐标，碰到相同的前缀和就删除这段数组，反复操作直到不能删除数组为止，最后生成链表。

class Solution:
    def removeZeroSumSublists(self, head: ListNode) -> ListNode:
        a = []
        while head:
            a += [head.val]
            head = head.next
        flag = True
        while flag:
            t = 0
            d = {0: -1}
            flag = False
            for i, v in enumerate(a) :
                t += v
                if t in d:
                    a = a[: d[t] + 1] + a[i + 1:]
                    flag = True
                    break
                d[t] = i
        ans = ListNode(0)
        t = ans
        for i in a:
            t.next = ListNode(i)
            t = t.next
        return ans.next

直接在链表上操作

class Solution:
    def removeZeroSumSublists(self, head: ListNode) -> ListNode:
        ans = ListNode(0)
        ans.next = head
        flag = True
        while flag:
            h = ans
            d = {0: ans}
            t = 0
            flag = False
            while h.next:
                h = h.next
                t += h.val
                if t in d:
                    d[t].next = h.next
                    flag = True
                    break
                d[t] = h
        return ans.next