51nod 1393 0和1相等串

题目大意

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

Input

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

Output

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

Input示例

1011

Output示例

2

分析

把0当成-1，然后求前缀和。如果出现前缀和相等的，那么这个串肯定就满足要求。因为只要求最长的，所以只需要存前缀和首次出现的位置即可，之后再遇到就可以直接更新答案。

代码

#include <bits/stdc++.h>#define N 2000010int sum[N];char ch[N * 2];int main(){    gets(ch);    memset(sum,-1,sizeof(sum));    sum[strlen(ch)] = 0;    int ans = 0;    int last = strlen(ch);    for (int i = 0; i < strlen(ch); i++)    {        if (ch[i] == '0')            last--;            else last++;        if (sum[last] > -1)            ans = std::max(ans, i - sum[last] + 1);            else sum[last] = i + 1;    }    printf("%d\n",ans);}