51nod 1393 0和1相等串

题目链接：https://www.51nod.com/onlineJudge/questionCode.html#!problemId=1393

题目：

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

Input

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

Output

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

Input示例

1011

Output示例

2

有个O（n）算法。

#include <iostream>
#include<bits/stdc++.h>
#define N 1100000
using namespace std;

int v[2*N];
char s[2*N];

int main()
{
scanf("%s",s);
int len=strlen(s);
int sum=N,ans=0;
for(int i=0;i<len;i++)
{
sum+=(s[i]=='0'?-1:1);
if(sum==N) ans=i+1;
else if(!v[sum]) v[sum]=i+1;
else ans=max(ans,i+1-v[sum]);
}
cout<<ans<<endl;
}