1393 0和1相等串

1393 0和1相等串

基准时间限制：1 秒 空间限制：131072 KB 

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

Input

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

Output

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

Input示例

1011

Output示例

2
思路：把0变成-1，然后01相等，那个段的和就是0，然后维护前缀和，当前缀和为0的时候，我们更新最长的，那么当当前前缀不是0的时候，如果我们知道最前面的和
等于当前值的时候（j），那么[j+1，i]的和就为0，那么当前值和最大比较更新。复杂n*log(n);

#include<stdio.h>
#include<algorithm>
#include<iostream>
#include<string.h>
#include<queue>
#include<math.h>
#include<set>
#include<vector>
#include<string.h>
#include<map>
using namespace std;
typedef long long LL;
int sum[1000005];
map<int,int>my;
char str[1000005];
int ans[1000005];
int main(void)
{
    my.clear();int i , j;
    scanf("%s",str);
    int l= strlen(str);
    for(i = 0;i < l; i++)
    {
        if(str[i] == '0')
        {
            ans[i+1] = -1;
        }
        else ans[i+1] = 1;
    }
    my[0] = 0;int maxx = 0;
    sum[0] = 0;
    for(i = 1; i <= l; i++)
    {
        sum[i] = sum[i-1] + ans[i];
        if(sum[i] == 0)
        {
            maxx = max(i,maxx);
        }
        else
        {
            if(my[sum[i]]==0)
            {
                my[sum[i]] = i;
            }
            else maxx = max(i-my[sum[i]],maxx);
        }
    }
    printf("%d\n",maxx);
    return 0;
}