2439 halobin 的序列

题目描述

Halobin有一个01序列，他每次可以翻转一个元素，即将该元素从0变1或者从1变0。

现在他希望序列不降，求最少翻转次数。

输入

第一行输入一个数n，表示字符串的长度。其中1≤n≤20000；
第二行输入一个由‘0’和‘1’组成的字符串

输出

输出一个非负整数，表示最少翻转次数

样例输入

6
010110

样例输出

2

提示

 

题型： 前缀和，后缀和

不下降的 01 序列只有三类：

1、00000…00000

2、11111…11111

3、000…0011…111

因此可以考虑把原序列前 x 个数全改成 0，后 n-x 个数全改成 1，通过枚举 x=1…n 来计算代

价。

暴力枚举 x 的位置

暴力计算把 a[1]~a[x]全部置为 0 的代价，这部分代价就是前 x 个数中 1 的个数

暴力计算把 a[x+1]~a[n]全部置为 1 的代价，这部分代价就是后 n-x 个数中 0 的个数

两部分代价取和，不断维护最小值

不难看出，从 1…n 枚举 x 的值，就能覆盖刚刚讨论的三种情况了。

于是我们有了一个 O(n^2)的做法。

通过刚才的分析，可以看出我们只要预先处理出来前缀中 1 的个数和后缀中 0 的个数，就能

够 O(1)计算每一个所对应的答案了。

通过前缀和优化，我们将算法的复杂度优化到 O(n)了。

 

#include <bits/stdc++.h>
using namespace std;
const int N = 3e5 + 10;

int n;
char s[N];
// i Œª÷√«∞√Ê”–∂‡…Ÿ∏ˆ1, iŒª÷√∫Û”–∂‡…Ÿ∏ˆ0 
int pre[N], last[N];
//暴力枚举+前缀和优化。  O(n^2)==>O(n)
int main()
{
	cin >> n;
	for (int i=1; i<=n; i++)
		cin >> s[i];

	for (int i=1; i<=n; i++) {
		pre[i] = pre[i-1];//统计前面1的数量
		if (s[i] == '1')
			pre[i] ++;
	} 
	for (int i=n; i>=1; i--) {
		last[i] = last[i+1];//统计后面0的数量
		if (s[i] == '0')
			last[i]++;
	}
	int res = n + 5;
	for (int i=1; i<=n; i++) {
		res = min(res, pre[i-1] + last[i+1]);//如果i前面都是0，i后面都是1以后，i本身是0还是1与最后结果无关
//i与结果无关，看0-(i-1)和(i+1)-(n-1)就可以了
	}
	cout << res << endl;
	return 0;
}