C++のリンクリストからゼロサム連続ノードを削除する

リンクリストの先頭を指定したとします。そのようなシーケンスがなくなるまで、合計が0になるノードの連続するシーケンスを繰り返し削除する必要があります。したがって、そうした後、最終的なリンクリストの先頭を返す必要があります。したがって、リストが[1,2、-3,3,1]のような場合、結果は[3,1]になります。

これを解決するには、次の手順に従います-

• ダミーと呼ばれるノードを作成し、そこに0を格納し、ダミーの次を設定します：=head

• 1つのマップmを作成し、キー0のダミーをmに格納し、sum=0に設定します

• 頭がヌルではない間-

  • sum：=sum +ヘッドの値、m [sum]：=ヘッド、およびヘッド：=ヘッドの次を設定

• 頭：=ダミー

• 合計：=0

• 頭がヌルではない間

  • 合計：=合計+ヘッドの値

  • temp：=m [sum]

  • tempがheadでない場合は、next of head：=next of temp

  • 頭：=頭の次

• ダミーの次を返す

理解を深めるために、次の実装を見てみましょう-

例

#include <bits/stdc++.h>
using namespace std;
class ListNode{
   public:
   int val;
   ListNode *next;
   ListNode(int data){
      val = data;
      next = NULL;
   }
};
ListNode *make_list(vector<int> v){
   ListNode *head = new ListNode(v[0]);
   for(int i = 1; i<v.size(); i++){
      ListNode *ptr = head;
      while(ptr->next != NULL){
         ptr = ptr->next;
      }
      ptr->next = new ListNode(v[i]);
   }
   return head;
}
void print_list(ListNode *head){
   ListNode *ptr = head;
   cout << "[";
   while(ptr){
      cout << ptr->val << ", ";
      ptr = ptr->next;
   }
   cout << "]" << endl;
}
class Solution {
   public:
   ListNode* removeZeroSumSublists(ListNode* head) {
      ListNode* dummy = new ListNode(0);
      dummy->next = head;
      unordered_map <int, ListNode*> m;
      m[0] = dummy;
      int sum = 0;
      while(head){
         sum += head->val;
         m[sum] = head;
         head = head->next;
      }
      head = dummy;
      sum = 0;
      while(head){
         sum += head->val;
         ListNode* temp = m[sum];
         if(temp != head){
            head->next = temp->next;
         }
         head = head->next;
      }
      return dummy->next;
   }
};
main(){
   vector<int> v1 = {1,2,-3,3,1};
   ListNode *head = make_list(v1);
   Solution ob;
   print_list(ob.removeZeroSumSublists(head));
}

入力

[1,2,-3,3,1]

出力

[3,1]