力扣 1171. 从链表中删去总和值为零的连续节点 链表+暴力/哈希

https://leetcode-cn.com/problems/remove-zero-sum-consecutive-nodes-from-linked-list/comments/

思路一：暴力。枚举起始点$l$，找到一个终结点$r$，使得$[l,r)$元素之和等于$0$，然后修改节点之间的关系即可。最差情况下时间复杂度为$O(n^2)$。注意此题不能使用$delete$回收空间。

/**
 * 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) {
        ListNode myhead(0);
        myhead.next=head;
        ListNode *pre=&myhead,*l=head,*r;
        while(l){
            r=l->next;
            int val=l->val;
            while(val&&r){
                val+=r->val;
                r=r->next;
            }
            if(!val){
                l=r;
                pre->next=l;
            }
            else
                pre=l,l=l->next;
        }
        pre=myhead.next;
        return myhead.next;
    }
};

思路二：利用哈希，$m[val]=head$表示从伪头节点(自己建立一个值为0的节点)到$head$的前缀和为$val$。先遍历一遍链表，计算出这个哈希表(同一键多次出现，采用覆盖策略)。然后再次遍历一遍链表，修改$cur->next$为$m[val]->next$即可。为什么这样可行呢？因为当我们第二次遍历链表至节点$l$时，已知$val$是链表头到$l$的前缀和，那么依据之前的计算规则，若$m[val]=r$，则$r$要么等于$l$，要么位于$l$之后，且从链表头到$r$的前缀和也为$val$，那我们自然可以删去$[l+1,r]$这一段。无论哪一种情况都符合我们的操作，所以结果是正确的。

/**
 * 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) {
        ListNode myhead(0);
        myhead.next=head;
        unordered_map<int,ListNode*> m;
        int val=0;
        ListNode *cur=&myhead;
        while(cur){
            val+=cur->val;
            m[val]=cur;
            cur=cur->next;
        }
        val=0;
        cur=&myhead;
        while(cur){
            val+=cur->val;
            cur->next=m[val]->next;
            cur=cur->next;
        }
        return myhead.next;
    }
};