leetcode 1171. 从链表中删去总和值为零的连续节点 前缀和+哈希表+链表

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.

通过次数9,209
提交次数20,320

---

• 第一遍遍历，用哈希表标记前缀和，
• 第二遍遍历，用哈希表记录好的位置来跳过中间区间和为0的部分
• 题目好像有点小问题，见代码注释

/**
 * Definition for singly-linked list.
 * public class ListNode {
 *     int val;
 *     ListNode next;
 *     ListNode(int x) { val = x; }
 * }
 */
class Solution {
    public ListNode removeZeroSumSublists(ListNode head) {
        Map<Integer, ListNode> mp = new HashMap<>();
        int sum = 0;
        ListNode ptr = head;
        mp.put(0, new ListNode(0));
        mp.get(0).next = head;
        for(ptr = head; ptr != null; ptr = ptr.next) {
            sum += ptr.val;
            mp.put(sum, ptr);
        }
        ptr=mp.get(0); 
        ptr.val = 0; //这里要赋值为0不然报错？
        sum=0;
        for(; ptr!=null; ptr=ptr.next) {
            sum += ptr.val;
            ptr.next = mp.get(sum).next;
        }
        return mp.get(0).next;
    }
}