51Nod 1391 01串(思维+前缀和)

传送门

题意：

思路：预处理出每个点向左和向右分别最远能扩展多远。处理完后O(n)扫一遍串即可。

很常规的0->-1,1->1维护一个【0,i】的前缀和cur，当cur<0表示[0,i]区间0的个数大于1的个数，即当前长度为最长的，否则判断cur+1是否出现过，如果cur+1这个数值没有出现过，则向左扩展最大长度为0，如果曾经出现过，则两点间的串0比1多，因为有前缀和可知，若两点前缀和相同，都为x，则中间那段加和为0，若存在x+1，则相当于前边那个位置后移了1为，且那一位为1，所以加和是x+1，中间段就少个1，0比1多。易知cur+1是最优的，因为cur+2、cur+3必定后出现。

#include<iostream>
#include<cstring>
using namespace std;
const int maxn = 1e6+5;
int pos[maxn],l[maxn],r[maxn];
void gmax(int &a,int b){
    if(a<b) swap(a,b);
}
char s[maxn];
int main(){
    scanf("%s",s);
    int cur = 0,len = strlen(s);
    memset(pos,-1,sizeof(pos));
    for(int i=0;i<len;i++){
        if(s[i]=='1') cur += 1;
        else cur += -1;
        if(cur<0) l[i] = i+1;
        else {
            if(pos[cur+1]!=-1) l[i] = i-pos[cur+1];
            else pos[cur] = i;
        }
    }
    cur = 0;
    memset(pos,-1,sizeof(pos));
    for(int i=len-1;i>=0;i--){
        if(s[i]=='1') cur+=1;
        else cur+=-1;
        if(cur>0) r[i]=len-i;
        else {//cur<0!!!
            if(pos[-(cur-1)]!=-1) r[i] = pos[-(cur-1)]-i;
            else pos[-cur] = i;
        }
    }
    int ans = 0;
    for(int i=0;i<len;i++){
        if(l[i] && r[i+1]) gmax(ans,l[i]+r[i+1]);
    }
    printf("%d\n",ans);
    return 0;
}