CodeForces - 873B - Balanced Substring

题目链接：http://codeforces.com/problemset/problem/873/B

题意：一个01串如果它的0的个数与1的个数相同，那么他就是一个平衡串，现在给出一个01串，问它的最长平衡子串的长度。

解析：用权值前缀的思想，将"0"的的权值看做-1，将"1"的权值看做1，对题中所给01串求前缀权值和，那么到第i个字符的权值前缀和bit[i].val代表他前面出现的"0"和"1"的权值和为bit[i].val，那么对于权值前缀和相同的两个字符串str[i],str[j]，容易发现stri+1，stri+2，stri+3...strj这个子串中的"0"的个数与"1"的个数相同。那么我们按"权值前缀和"排序后检查此值相同的元素之间的最大下标差即可。

代码：

#include <bits/stdc++.h>
using namespace std;
const int M=100005;
int n;
char str[M];
struct Node{
    int val,id;
    bool operator <(const Node & obj)const
    {
        return (val<obj.val)||((val==obj.val)&&(id<obj.id));
    }
}bit[M];
int main()
{
    scanf("%d",&n);
    scanf("%s",str+1);
    bit[1].val=0;
    bit[1].id=1;
    for(int i=2;i<=n+1;i++)
    {
        int v=str[i-1]=='0'?-1:1;
        bit[i].val=bit[i-1].val+v;
        bit[i].id=i;
    }
    sort(bit+1,bit+n+2);
    int ans=0;
    for(int i=1;i<=n+1;i++)
    {
        int tval=bit[i].val;
        int tid=bit[i].id;
        while(i<=n&&(bit[i+1].val==tval)) i++;
        ans=max(ans,bit[i].id-tid);
    }
    printf("%d\n",ans);
    return 0;
}