將C++陣列中所有零移到開頭,所有一移到結尾
在本教程中,我們將編寫一個程式,將所有零移到陣列的開頭,並將所有一移到陣列的結尾。
給定一個包含零、一和隨機整數的陣列。我們需要將所有零移到陣列的開頭,並將所有一移到陣列的結尾。讓我們來看一個例子。
輸入
arr = [4, 5, 1, 1, 0, 0, 2, 0, 3, 1, 0, 1]
輸出
0 0 0 0 4 5 2 3 1 1 1 1
演算法
初始化陣列。
將索引初始化為**1**。
迭代給定的陣列。
如果當前元素不是零,則使用當前元素更新索引處的數值。
遞增索引。
編寫一個迴圈,從上面的索引迭代到**n**
將所有元素更新為**1**。
對**0**也同樣進行操作。不用增加索引,而是減小索引,將所有零移到陣列的開頭。
實現
以下是上述演算法在C++中的實現
#include <bits/stdc++.h> using namespace std; void update1And0Positions(int arr[], int n) { int index = 0; for (int i = 0; i < n; i++) { if (arr[i] != 1) { arr[index++] = arr[i]; } } while (index < n) { arr[index++] = 1; } index = 0; for (int i = n - 1; i >= 0; i--) { if (arr[i] == 1) { continue; } if (!index) { index = i; } if (arr[i] != 0) { arr[index--] = arr[i]; } } while (index >= 0) { arr[index--] = 0; } } int main() { int arr[] = { 4, 5, 1, 1, 0, 0, 2, 0, 3, 1, 0, 1 }; int n = 12; update1And0Positions(arr, n); for (int i = 0; i < n; i++) { cout << arr[i] << " "; } cout << endl; return 0; }
輸出
如果執行以上程式碼,則會得到以下結果。
0 0 0 0 4 5 2 3 1 1 1 1
廣告