题意:给出一个字符串,求其中满足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;
}