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

目录链接：

力扣编程题-解法汇总_分享+记录-CSDN博客

GitHub同步刷题项目：

https://github.com/September26/java-algorithms

原题链接：力扣

描述：

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

解题思路：

* 解题思路：

* 构建前缀和，相同前缀和的位置，其连续累加值必定为0，相同前缀和之间的节点都是可以被删掉的。

* 另外，如果有多个前缀和都相同的节点，我们只需要取第一个和最后一个即可。

* 所以构建一个map，key为前缀和，value则为对应的节点。

* 首先遍历第一遍，找到最后一个前缀和的节点加入到map中。

* 然后第二遍循环，找到等于当前前缀和值的节点，这时候是一定存在的，是不过有可能是当前的节点。

* 如果是当前节点的话，使用node->next = seen[prefix]->next;也满足要求。

代码：

class Solution1171
{
public:
    ListNode *removeZeroSumSublists(ListNode *head)
    {
        ListNode *newHead = new ListNode(0);
        newHead->next = head;
        int prefix = 0;
        map seen;
        for (ListNode *node = newHead; node; node = node->next)
        {
            prefix += node->val;
            seen[prefix] = node;
        }
        prefix = 0;
        for (ListNode *node = newHead; node; node = node->next)
        {
            prefix += node->val;
            node->next = seen[prefix]->next;
        }
        return newHead->next;
    }
};