基准时间限制:1 秒 空间限制:131072 KB 分值: 20
难度:3级算法题
给定一个0-1串,请找到一个尽可能长的子串,其中包含的0与1的个数相等。
Input
一个字符串,只包含01,长度不超过1000000。
Output
一行一个整数,最长的0与1的个数相等的子串的长度。
Input示例
1011
Output示例
2
原来前缀和的功能那么强大,我们把0当作-1,那么要是区间内【i,j】0和1的个数相等,要么前缀和为0,要么【1,i-1】的前缀和等于【1,j】的前缀和,我们可以用map记录前缀和的下标,每次更新,得到答案即可。
#include<iostream>
#include<algorithm>
#include<string.h>
#include<map>
using namespace std;
map<int,int> mp;
#define MAXN 1000500
char c[MAXN];
char rc[MAXN];
int sum[MAXN];
int main()
{
cin>>c;
int l=strlen(c);
memset(sum,0,sizeof(sum));
for(int i=1;i<=l;i++)
{
rc[i]=c[i-1];
}
for(int i=1;i<=l;i++)
{
if(rc[i]=='1') sum[i]=sum[i-1]+1;
else sum[i]=sum[i-1]-1;//记录前缀和
}
int ans=0;
for(int i=1;i<=l;i++)
{
if(!sum[i])
{
ans=i;//前缀和为0
}
if(!mp[sum[i]])
{
mp[sum[i]]=i;//记录第一次前缀和为某值的下标,因为是最早的,所以map[sum[i]]的值不必更新
}
if(mp[sum[i]])
{
ans=max(ans,i-mp[sum[i]]);//更新区间 (前缀和相等的时候)
}
}
cout<<ans<<endl;
return 0;
}