51nod 1393 0和1相等串

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

Input

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

Output

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

Input示例

1011

Output示例

2
下标法，在cf遇到相同的题。。。。

#include <iostream>
using namespace std;
#include<string.h>
#include<set>
#include<stdio.h>
#include<math.h>
#include<queue>
#include<map>
#include<algorithm>
#include<cstdio>
#include<cmath>
#include<cstring>
#include <cstdio>
#include <cstdlib>
#include<stack>
#include<vector>
const int MAX=1000000;

char a[2*MAX+10];
int b[2*MAX+10];
int main()
{
    cin>>a+2;
    int len=strlen(a+2);
    int sum=0;
    int max1=0;
    memset(b,0,sizeof(b));
    b[MAX]=1;
    for(int i=2;i<=len+1;i++)
    {
        if(a[i]=='1')
            sum++;
        else
            sum--;
        if(b[sum+MAX]!=0)
            max1=max(max1,i-b[sum+MAX]);
        else
            b[sum+MAX]=i;
    }
    cout<<max1<<endl;
    return 0;
}