LeetCode 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]

提示：

• 给你的链表中可能有 1 到 1000 个节点。
• 对于链表中的每个节点，节点的值：-1000 <= node.val <= 1000.

思路

遍历时记录前缀和及其位置，当某个前缀和重复出现时，两次出现之间的节点（不包括第一次出现的那个节点）都可以删掉，由于head也可以被删掉，所以在head之前再加一个节点。

代码

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
class Solution {
public:
    ListNode* removeZeroSumSublists(ListNode* head) {
        bool ok = false;
        ListNode *header = new ListNode(0);
        
        header->next = head;
        
        while (!ok) {
            int sum = 0;
            unordered_map<int, ListNode*> sum2node;

            ok = true;

            for (auto *iter = header; iter != nullptr; iter = iter->next) {
                sum += iter->val;
                if (sum2node[sum] != nullptr) {
                    ok = false;

                    int tmpSum = sum;
                    auto left = sum2node[sum], right = iter->next;
                    for (auto iter = left->next; iter != right; iter = iter->next)
                        sum2node[tmpSum += iter->val] = nullptr;

                    left->next = right;
                    sum2node[sum] = left;
                } else sum2node[sum] = iter;
            }
        }

        return header->next;
    }
};