作业训练五 编程题 2. 0-1串

作业训练五 编程题 2. 0-1串

【问题描述】
对于一个包含 n 个整数元素的序列a1、a2、…、an，每个元素的值或者是0或者是1，选择两个下标i和j（1≤i≤j≤n）,对于所有的此范围内的元素ak（i≤k≤j），执行操作ak =1- ak。

选择合适的i和j，执行上述操作一次之后，可以得到的新序列中包含1的个数最多是多少？

【输入形式】
输入的第一行为一个整数 n（1≤ n ≤100），接来的一行为 n 个整数，每个整数或者是0或者是1。

【输出形式】
输出为一个整数，表示执行一次上述操作后可以获得的最大1的个数。

【样例输入1】

5
1 0 0 1 0
【样例输出1】

4
【样例输入2】

4
1 0 0 1
【样例输出2】

4
【样例说明】

在第一个样例中，选择i=2, j=5, 改变后的序列为[1 1 1 0 1]，包含4个1，很显然无法改变为[1 1 1 1 1]。

在第二个样例中，选择i=2, j=3，改变后的序列为[1 1 1 1]，包含4个1。

//把0全部换为-1，得到元素和最小的子序列，即0于1的个数差最大
//该子序列的1和-1互换

#include <iostream>
#include <vector>
#include <limits.h>
using namespace std;
int main() {
	int n = 0;
	cin >> n;
	vector<int> vec(n);
	int num1 = 0;

	//输入序列，转化为-1和1
	int temp = 0;
	for (int i = 0; i < n; i++) {
		cin >> temp;
		if (temp == 0) {
			vec[i] = -1;
		}
		else {
			vec[i] = 1;
			num1++;
		}
	}

	//动态规划+前缀和
	//对于每个元素， dp[i]存放其作为子序列尾部时，
	//包含-1与1之和最小的值
	vector<int> dp(n, 0);
	dp[0] = vec[0];
	for (int i = 1; i < n; i++) {
		if (dp[i - 1] < 0) {
			dp[i] = dp[i - 1] + vec[i];
		}
		else dp[i] = vec[i];
	}


	//找出最小值,并求出将此序列调换之后1的个数
	int min = INT_MAX;
	for (int i = 0; i < n; i++) {
		if (min > dp[i]) min = dp[i];
	}
	num1 -= min;
	cout << num1 << endl;
}