51nod-【1393 0和1相等串】

1393 0和1相等串

基准时间限制：1 秒 空间限制：131072 KB 分值: 20 难度：3级算法题

 收藏

 关注

给定一个0-1串，请找到一个尽可能长的子串，其中包含的0与1的个数相等。

Input

一个字符串，只包含01，长度不超过1000000。

Output

一行一个整数，最长的0与1的个数相等的子串的长度。

Input示例

1011

Output示例

2

网上有各种写法，其实题意让求出一个子串0,1个数相同，

我们可以把0看成-1，求出前缀和，记录每一个前缀和的位置 

我们只需要把前缀和相同的相减求出最大值就行了

举个栗子： 1 2 4……44  55

看这个序列，如果2-44中间存在相加等于0的数那么前缀和相加等于1，

就会出现2次，我们只需要把位置相减就是其长度

#include<cstdio>#include<cstring>#include<algorithm>using namespace std;struct node{int sum;int pos;}a[1000000+11];char s[1000000+11];bool cmp(node d,node c){if(d.sum!=c.sum)return d.sum<c.sum;return d.pos<c.pos; } int main(){scanf("%s",s);int i,k,len=strlen(s);a[0].sum=0;a[0].pos=0;k=1;//对于a[0].pos,a[0].sum,一定要初始化开始没有把a[0].pos初始为0//对于011这样的字符我的输出为0，不懂为什么思考一下 for(i=0;i<len;++i){if(s[i]=='0'){a[k].sum=a[k-1].sum-1;a[k].pos=k; }else{a[k].sum=a[k-1].sum+1;a[k].pos=k; } ++k; }/*for(i=0;i<k;++i)printf("%d ",a[i].sum);printf("\n");*/ sort(a,a+k,cmp);/*for(i=0;i<k;++i)printf("%d %d\n",a[i].sum,a[i].pos);*/int dis,mmax=0;for(i=1;i<k-1;++i){if(a[i].sum==a[i+1].sum){int temp=a[i].pos;while(i<k-1&&a[i].sum==a[i+1].sum)++i;dis=a[i].pos-temp;if(dis>mmax) mmax=dis;} }printf("%d\n",mmax); return 0;}