C++ですべてのゼロをリンクリストの先頭に移動します

ランダムな整数とゼロを持つリンクリストが与えられます。すべてのゼロをリンクリストの先頭に移動する必要があります。例を見てみましょう。

入力

3 -> 0 -> 1-> 0 -> 0 -> 1 -> 0 -> 0 -> 3 -> NULL

出力

0->0->0->0->0->3->1->1->3->NULL

アルゴリズム

• リンクリストを初期化します。
• リンクリストが空であるか、単一ノードである場合に戻ります。
• 2つのノードをそれぞれ2番目のノードと最初のノードで初期化して、現在のノードと前のノードを追跡します。
• リンクリストを最後まで繰り返します。
• 
  
  • 現在のノードが0の場合は、新しいヘッドにします。
  • 現在および以前のノード変数の値を更新します。
  • ノードを新しいヘッドにすると、ノードが前面に移動します。
  • 新しいヘッドの次の値を前のヘッドで更新します。

実装

以下は、C++での上記のアルゴリズムの実装です

#include <bits/stdc++.h>
using namespace std;
struct Node {
   int data;
   struct Node *next;
};
void addNewNode(struct Node **head, int data) {
   struct Node *newNode = new Node;
   newNode->data = data;
   newNode->next = *head;
   *head = newNode;
}
void moveZeroes(struct Node **head) {
   if (*head == NULL) {
      return;
   }
   struct Node *temp = (*head)->next, *prev = *head;
   while (temp != NULL) {
      if (temp->data == 0) {
         Node *current = temp;
         temp = temp->next;
         prev->next = temp;
         current->next = *head;
         *head = current;
      }else {
         prev = temp;
         temp = temp->next;
      }
   }
}
void printLinkedList(struct Node *head) {
   while (head != NULL) {
      cout << head->data << "->";
      head = head->next;
   }
   cout << "NULL" << endl;
}
int main() {
   struct Node *head = NULL;
   addNewNode(&head, 3);
   addNewNode(&head, 0);
   addNewNode(&head, 1);
   addNewNode(&head, 0);
   addNewNode(&head, 0);
   addNewNode(&head, 1);
   addNewNode(&head, 0);
   addNewNode(&head, 0);
   addNewNode(&head, 3);
   moveZeroes(&head);
   printLinkedList(head);
   return 0;
}

出力

上記のコードを実行すると、次の結果が得られます。

0->0->0->0->0->3->1->1->3->NULL