51nod1393 0和1相等串 思维题

题目：

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

题意：

思路：

很好的思维题。维护一个前缀和，把0看做-1，一直加，记录每种前缀和首次出现的位置，当再次得到这个前缀和，说明当前位置和首次出现位置之间0和1的个数一定相等，更新答案。因为直接前缀和有负数，无法用数组下标直接记录，所以统一加一个len

#include 

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;
}