CodeFroces 873B. Balanced Substring(思维题)

点击打开链接

题意：给出一个字符串，求其中满足0和1的个数相等的最长字符串的长度。

解法：当时只能想到枚举区间然后直接用前缀和判定的方法，但是枚举区间复杂度明显不对。

然后想到了二分，发现很多数据都挺对的，然后wa24。想了想发现，有可能6不满足但是8满足（00111100），故不存在单调性。。。

最后看了一发题解。每位有一个值：前i个数字0的个数 - 前i个数字1的个数。如果有两个地方的值是相同的，那么显然，他们包围的区间是满足条件的。也很显然，你肯定是要某个数值最左边的坐标和最右边的坐标，这样才是最大的。然后取满足条件的最大区间即可。

有一个细节的地方需要注意，就是数值为0的地方，否则wa12。

代码如下：

#include<iostream>
#include<cstdio>
#include<vector>
#include<queue>
#include<utility>
#include<stack>
#include<algorithm>
#include<cstring>
#include<string>
#include<cmath>
#include<set>
#include<map>
using namespace std;
const int maxn = 1e5 + 5;
int n, cnt0[maxn], cnt1[maxn], cnt[maxn * 3];
int LEFT[maxn * 3], RIGHT[maxn * 3];
char str[maxn];
bool vst[maxn * 3];

int main() {
#ifndef ONLINE_JUDGE
	freopen("in.txt", "r", stdin);
//    freopen("out.txt", "w", stdout);
#endif
	scanf("%d", &n);
	scanf("%s", str + 1);
	vst[maxn] = 1;
	LEFT[maxn] = RIGHT[maxn] = 0;
	for(int i = 1; str[i]; i++) {
		cnt[i] = cnt[i - 1];
		if(str[i] == '0')
			cnt[i]++;
		else
			cnt[i]--;
		if(!vst[cnt[i] + maxn]) {
			vst[cnt[i] + maxn] = 1;
			LEFT[cnt[i] + maxn] = RIGHT[cnt[i] + maxn] = i;
		} else {
			RIGHT[cnt[i] + maxn] = i;
		}
	}
	int ans = 0;
	for(int i = 0; i < maxn * 3; i++) {
		if(!vst[i])
			continue;
		ans = max(ans, RIGHT[i] - LEFT[i]);
	}
	printf("%d\n", ans);
	return 0;
}