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

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

题目

给你一个链表的头节点 head，请你编写代码，反复删去链表中由 总和 值为 0 的连续节点组成的序列，直到不存在这样的序列为止。

删除完毕后，请你返回最终结果链表的头节点。

你可以返回任何满足题目要求的答案。

示例

示例 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]

解题

思路

把每一个节点的值和它前面的节点的值得和保存在一个哈希表中，键为和的值，值为当前节点，当有一个节点和之前的和在哈希表中有的话,那就说明，这个节点之前(包括这个节点)到有那个节点之后的节点之和等于0，把这段链表删除即可。

为了便于统计头节点，所以我们还是需要虚拟头节点dummy。

代码

class Solution {
    public ListNode removeZeroSumSublists(ListNode head) {
        ListNode fake = new ListNode(0);
        fake.next = head;
        Map<Integer, ListNode> map = new HashMap<>();
        map.put(0, fake);
        Integer preSum = 0;

        while (head != null) {
            preSum = preSum + head.val;
            if (map.get(preSum) == null) {
                map.put(preSum, head);
            } else {
                map.get(preSum).next = head.next;
            }
            head = head.next;
        }

        return fake.next;
    }
}