给定一个0和1的数组。任务是找到将数组中所有1分组在一起所需的最小交换次数。
如果输入数组= {1、0、1、1、0、1},则需要进行1次交换。即将前0与后1交换。
计算数组中1的总数
如果count是x,那么我们需要找到该数组的长度为x的子数组,其最大数目为1
所需的最小交换数将是长度为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