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

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

这题解法很奇妙。也是看了别人的思路才想出来.

首先考虑把0全部变成-1。然后统计一个前缀和,用sum[i]表示.

那么从起点开始的子串是合法的只要sum[i]的值等于0即可.

如果子串起点不为0,那么只要两次出现sum[i],这两个数值之间一定是合法的子串.

例如:

  标号：   1  2  3  4  5  

  数值      0  0  1  1  0  

  sum     -1 -2 -1 0  -1

i==4的时候sum[i]==0,所以1-4之间是一个合法的子串,长度为4  

sum[1]==sum[3]==sum[5]== -1,所以 (1,3],(1,5],(3,5]之间都是合法的子串,取最大值也是4.

我是用了map,其实不用这么麻烦,只要用一个变量就可以代替sum数组,并且另外开一个标记数组可以代替map做映射.

#include <iostream>
#include <string.h>
#include <stdio.h>
#include <map>
using namespace std;
char s[1000010];
int t[1000010],sum[1000010];
int main()
{
    //freopen("a.txt","r",stdin);
    scanf("%s",s+1);
    int l=strlen(s+1);
    for(int i=1;i<=l;i++)
        if(s[i]=='0') t[i]=-1;
        else t[i]=1;
    //for(int i=0;i<l;i++) printf("%d",t[i]);
    sum[1]=t[1];
    map<int,int>m;
    m[sum[1]]=1;
    int maxn=0;
    for(int i=2;i<=l;i++)
    {
        sum[i]=sum[i-1]+t[i];
        //printf("%d
",sum[i]);
        if(m[sum[i]]!=0) maxn=max(maxn,i-m[sum[i]]);
        else m[sum[i]]=i;
        if(sum[i]==0) maxn=max(maxn,i);
    }
    printf("%d
",maxn);
    return 0;
}