51nod1393 0和1相等串 思維題

題目：

http://www.51nod.com/onlineJudge/questionCode.html#!problemId=1393

題意：

思路：

很好的思維題。維護一個前綴和，把0看做-1，一直加，記錄每種前綴和首次出現的位置，當再次得到這個前綴和，說明當前位置和首次出現位置之間0和1的個數一定相等，更新答案。因爲直接前綴和有負數，無法用數組下標直接記錄，所以統一加一個len

#include <bits/stdc++.h>

using namespace std;

const int N = 1000000 + 10;

char str[N];
int vis[N*2];

int main()
{
    scanf("%s", str + 1);
    int len = strlen(str + 1);
    int val = len, ans = 0;
    memset(vis, -1, sizeof vis);
    vis[len] = 0;//前綴和爲0首次出現的位置爲0
    for(int i = 1; i <= len; i++)
    {
        if(str[i] == '0') val--;
        else val++;
        if(vis[val] == -1) vis[val] = i;
        else ans = max(ans, i - vis[val]);
    }
    printf("%d\n", ans);
    return 0;
}