1171 從連結串列中刪去總和值為零的連續節點
阿新 • • 發佈:2021-02-07
技術標籤:LeetCode
題目描述:
給你一個連結串列的頭節點 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 到 1000 個節點。
對於連結串列中的每個節點,節點的值:-1000 <= node.val <= 1000.
方法1:
主要思路:解題連結彙總
(1)字首和,將之前出現過的和作為unordered_map<int,ListNode*> mp的鍵,儲存對應的結點地址;
(2)當同一個字首和再次出現,說明這兩個位置之前的和是0,故可以刪除這兩個位置之前的值;
(3)初始化時,加一個輔助頭結點;
/**
* 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) {
unordered_map<int,ListNode*> mp;
ListNode* dummy=new ListNode(0);//輔助頭結點
dummy->next=head;
mp[0]=dummy;//初始化
int cur_sum=0;
while(head){
cur_sum+=head->val; //字首和
if(mp.count(cur_sum)){//再次出現
ListNode* tmp=mp[cur_sum]->next;
mp[cur_sum]->next=head->next;//跳過兩個結點之間的結點
int v=cur_sum;
while(tmp!=head){//刪除對應的標識
v+=tmp->val;
tmp=tmp->next;
mp.erase(v);
}
}
else{//字首和的標識
mp[cur_sum]=head;
}
head=head->next;
}
head=dummy->next;
delete dummy;
return head;
}
};