在 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

更新日期:2019-12-20

206 次查看

开启您的 职场生涯

完成本课程后,即可获得认证

开始学习
广告
© . All rights reserved.