이 자습서에서는 모든 0을 배열의 맨 앞으로 이동하고 1을 끝으로 이동하는 프로그램을 작성할 것입니다.
임의의 정수와 함께 0과 1이 있는 배열이 제공됩니다. 모든 0을 시작으로 이동하고 1을 배열의 끝으로 이동해야 합니다. 예를 들어 보겠습니다.
입력
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로 초기화합니다.
-
주어진 배열을 반복합니다.
-
현재 요소가 0이 아니면 인덱스의 값을 현재 요소로 업데이트합니다.
-
인덱스를 증가시킵니다.
-
-
위의 인덱스에서 n까지 반복하는 루프를 작성하세요.
-
모든 요소를 1로 업데이트합니다.
-
-
마찬가지로 0에 대해 수행합니다. . 인덱스를 늘리는 대신 인덱스를 줄여 모든 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