基准时间限制: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; }