题目描述:
给定一个由 0 0 0和 1 1 1组成的数组,你必须选择一个非空区间,将该区间里面的元素进行翻转( 0 0 0变为 1 1 1, 1 1 1变为 0 0 0),问翻转后的数组最多有多少个 1 1 1。
题解;
我们很明显地发现,当翻转一个区间之后 1 1 1个数的增量为翻转区间里面 0 0 0的个数与 1 1 1的个数之差(如果两者之差为负数,那么此时翻转后 1 1 1的个数会减少),所以我们翻转的区间里面 0 0 0的个数一定要尽可能比 1 1 1的个数多,这样才能让 1 1 1增加的个数尽可能的多。
我们如何找到 0 0 0个数比 1 1 1个数尽可能多的区间呢?这就回到了最大连续子序列和问题,我们只需要将原数组中的 0 0 0记为 1 1 1,将原数组中 1 1 1记为 − 1 -1 −1,这样转换后一个区间的元素之和就是元素组对应区间中 0 0 0比 1 1 1多的个数(如果是负数,则代表原数组对应区间的 0 0 0更少),此时我们要找的就是转换后数组的最大连续子序列和,求最大连续子序列和有很多种方法,这里给出《算法导论》中给出的一种偏向数学的方法。
代码:
#include <bits/stdc++.h>
const int MAXLEN = 1e5 + 10;
using namespace std;
int n, ans, delta, nowSum, maxSum = -1;
int a[MAXLEN];
int main()
{
ios::sync_with_stdio(false);
cin >> n;
for (int i = 0; i < n; i++) {
cin >> a[i];
ans = ans + a[i]; // 记录原本1的个数
a[i] = a[i] == 0 ? 1 : -1; // 将0记为1,1记为-1
nowSum += a[i];
maxSum = max(maxSum, nowSum);
if (nowSum < 0) { nowSum = 0; }
}
ans += maxSum;
cout << ans << endl;
return 0;
}