1. 程式人生 > 其它 >1171 從連結串列中刪去總和值為零的連續節點

1171 從連結串列中刪去總和值為零的連續節點

技術標籤: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]

提示:
給你的連結串列中可能有 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; } };