Balanced Substring
前缀和思想下的暴力优化,正解应该是个DP
我们直接统计1的个数,进行前缀和,然后依次枚举区间长度,如果该区间的1的个数等于区间长度的一半,我们就记录这个值。
优化:
如果1的个数或者0的个数少于区间长度的一半,那么必然不成,直接退出循环。
通过观测可知,如果sum[r]-sum[l-1]满足条件要求,那么当l-1成为r的时候必然不存在新的l-1满足题目条件,若满足,这之前的sum[r]-sum[l-1]的最大长度会上升。所以开个vis数组记录用于加速。
#include<bits/stdc++.h>
using namespace std;
const int N=1e5+10;
char str[N];
int sum[N];
bool vis[N];
int main()
{
int n;
scanf("%d %s",&n,str+1);
for(int i=1;i<=n;++i)
{
sum[i]=sum[i-1];
if(str[i]=='1')
sum[i]++;
}
int maxx=0;
for(int i=n;i>=1;--i)
{
if(vis[i])continue;
int r=2;
while(i-r>=0)
{
if(r>sum[i]*2||r>(i-sum[i])*2)break;
if((sum[i]-sum[i-r])*2==r)
{
maxx=max(maxx,r);
vis[i-r]=true;
}
r+=2;
}
}
printf("%d\n",maxx);
}