C++ 中将二进制数组分区以使其先包含 0 后包含 1 的最小切换次数
问题陈述
给定一个仅包含 0 和 1 的 n 个整数数组。找到所需的最小切换次数(从 0 切换到 1 或反之亦然),使得数组被分区,即它首先包含 0,然后包含 1。
示例
如果 arr[] = {1, 0, 0, 1, 1, 1, 0},则需要 2 次切换,即切换第一个 1 和最后一个 0。
算法
- 如果我们观察问题,我们会发现从 0 到 n-1 肯定存在一个点,该点左侧的所有元素都应该包含所有 0,右侧的所有元素都应该包含所有 1。
- 不遵守此规则的索引将必须被移除。这个想法是从左到右计算所有 0。
示例
#include <bits/stdc++.h> using namespace std; int getMinToggles(int *arr, int n) { int zeroCnt[n + 1] = {0}; for (int i = 1; i <= n; ++i) { if (arr[i - 1] == 0) { zeroCnt[i] = zeroCnt[i - 1] + 1; } else { zeroCnt[i] = zeroCnt[i - 1]; } } int result = n; for (int i = 1; i <= n; ++i) { result = min(result, i - zeroCnt[i] + zeroCnt[n] - zeroCnt[i]); } return result; } int main() { int arr[] = {1, 0, 0, 1, 1, 1, 0}; int n = sizeof(arr) / sizeof(arr[0]); cout << "Minimum toggles = " << getMinToggles(arr, n) << endl; return 0; }
编译并执行上述程序时,它会生成以下输出:
输出
Minimum toggles = 2
广告