在 C++ 中將所有 1 組合到一起所需的最小交換次數
問題陳述
給定一個 0 和 1 的陣列。任務是找出將陣列中所有 1 組合在一起所需的最小交換次數。
例如
如果輸入陣列 = {1, 0, 1, 1, 0, 1},則需要 1 次交換。即交換第一個 0 和最後一個 1。
演算法
- 計算陣列中 1 的總數
- 如果計數為 x,那麼我們需要找出具有最大 1 個數量的長度為 x 的此陣列的子陣列
- 所需的最小交換次數將是長度為 x 的子陣列中 0 的數量,其中 1 的數量最大
例如
#include <bits/stdc++.h> using namespace std; int getMinSwaps(int *arr, int n) { int oneCnt = 0; for (int i = 0; i < n; ++i) { if (arr[i] == 1) { ++oneCnt; } } int x = oneCnt; int maxOnes = INT_MIN; int preCompute[n] = {0}; if (arr[0] == 1) { preCompute[0] = 1; } for (int i = 1; i < n; ++i) { if (arr[i] == 1) { preCompute[i] = preCompute[i - 1] + 1; } else { preCompute[i] = preCompute[i - 1]; } } for (int i = x - 1; i < n; ++i) { if (i == (x - 1)) { oneCnt = preCompute[i]; } else { oneCnt = preCompute[i] - preCompute[i - x]; } if (maxOnes < oneCnt) { maxOnes = oneCnt; } } int swapCnt = x - oneCnt; return swapCnt; } int main() { int arr[] = {1, 0, 1, 1, 0, 1}; int n = sizeof(arr) / sizeof(arr[0]); cout << "Minimum swap count = " << getMinSwaps(arr, n) << endl; return 0; }
當您編譯並執行上述程式時。它將生成以下輸出 −
輸出
Minimum swap count = 1
廣告