51nod-1391-01串（O（n）模拟）

1391 01串


题目来源： Codility

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


 收藏


 关注

给定一个01串S,求出它的一个尽可能长的子串S[i..j]，满足存在一个位置i<=x <j, S[i..x]中0比1多，而S[x + 1..j]中1比0多。求满足条件的最长子串长度。

Input

一行包含一个只由0和1构成的字符串S。 S的长度不超过1000000。

Output

一行包含一个整数，表示满足要求的最长子串的长度。

Input示例

10

Output示例

0

曹鹏 (题目提供者)

题解： 

  

首先需要预处理，01串最常见的预处理就是赋值，这里也不例外，将0看作-1，将1看作1

问题便转化成了前缀和小于0的最长长度与后缀和大于0的最长长度之和。

先从前往后遍历一遍，此时前缀和假如小于零，则标记长度即可

若此时前缀和大于零，则往前找到第一个出现前缀之和+1的地方，减去即可。

这里不难证明，第一次出现的情况比为当前情况的最优解

同理  后缀也一样。

#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
#define maxn 1000005
char s[maxn];
int a[maxn],b[maxn],c[maxn],h[maxn];
int  main()
{
int len,i,t=0,x,ans=0;
scanf("%s",s+1);
len=strlen(s+1);
for(i=1;i<=len;i++)
{
if(s[i]=='0')
a[i]=-1;
else
a[i]=1;
}
for(i=1;i<=len;i++)
{
t+=a[i];
if(t<0)
b[i]=i;
else
{
if(h[t+1]!=0)
b[i]=i-h[t+1];
else
{
h[t]=i;
b[i]=0;
}
}
}
t=0;
memset(h,0,sizeof(h));
for(i=len;i>0;i--)
{
t+=a[i];
if(t>0)
c[i]=len-i+1;
else
{
if(h[-t+1]!=0)
c[i]=h[-t+1]-i;
else
{
h[-t]=i;
c[i]=0;
}
}
}
for(i=1;i<=len;i++)
if(b[i]>0 && c[i+1]>0)
ans=max(ans,c[i+1]+b[i]);
printf("%d\n",ans);
}