51Nod 1393 0和1相等串 前缀和

51Nod 1393 0和1相等串 传送门

找含有的0和1数量相同的子串的长度……设一个前缀和sum的变量，遇到1加1，遇到0减1，sum=0时即为0和1数量相等的位置，一开始觉得只要没次sum=0重新开始计数，保存最大值就可以了，但是忽略了一种情况，当sum[i]==sum[j]时，i~j中间的0和1的数量也是相同的，长度为i-j；在处理前缀和时，可以开一个数组记录前缀和为x时的位置i+1，如果在sum！=0时，前缀和vis[sum]不为0，即已出现过sum，那么此时也可更新子串的长度max(ans,i+1-v[sum])

#include<iostream>
#include<algorithm>
#include<cmath>
#include<queue>
#include<vector>
#include<functional>
#include<string>
#include<string.h>
using namespace std;
typedef long long ll;
string s;
int v[1000005*2]; //不要忘了*2
int main()
{
    cin >> s;
    int ans = 0,sum=1e6;  //sum一开始要设置为1e6+，数组下标不为0
    memset(v,0,sizeof(v));
    for (int i = 0; i < s.length(); i++)
    {
        if (s[i] == '0') sum--;
        else  sum++;
        if (sum == 1e6) ans = i + 1;
        else if (!v[sum]) v[sum] = i + 1;
        else ans = max(ans, i + 1 - v[sum]);
    }
    cout << ans << endl;
    return 0;
}