题目大意
找一个最长的字串,满足字串中0和1的个数相同
思路
这里可以枚举做,将0看成-1,记录前缀和,那么我们只要找到一个最长的前缀和为0的字串就行了。但枚举也是有方法的,先枚举l,再从大的地方枚举r,r的减少也是有讲究的,不能每次减1,比如一个串”11111111”
在[1, 8]时前缀和之差sum[8] - sum[0] == 8,此时不满足,还要减少r来找到一个区间,使得sum[r] - sum[l - 1] == 0,这时候r怎么取呢?
很明显,前缀和之差为8,r至少要减8,之后的[l, r]才可能满足前缀和之差为0。因为r每减1,前缀和之差最多变化1。
#include <cstdio>
#include <algorithm>
#include <iostream>
#include <vector>
using namespace std;
typedef long long ll;
const int MAXN = 1000000 + 10;
const int NIL = -1;
const int INF = 0x3f3f3f3f;
char str[MAXN];
int sum[MAXN];
int main(void)
{
int n;
scanf("%d", &n);
scanf("%s", str);
sum[0] = 0;
for (int i = 1; i <= n; ++i)
{
sum[i] = sum[i - 1] + (str[i - 1] == '1' ? 1 : -1);
}
int ans = 0;
for (int l = 1; l <= n; ++l)
{
for (int r = n; r >= l; )
{
if (r - l + 1 < ans)
break;
int t = sum[r] - sum[l - 1];
if (t < 0)
t = -t;
if (t == 0)
{
ans = max(ans, r - l + 1);
break;
}
r -= t;
}
}
printf("%d\n", ans);
return 0;
}