51nod 1391 01串

来源:互联网 发布:js有集合吗 编辑:程序博客网 时间:2024/06/06 06:31

题目思路倒是不难,就是没有预处理出来。这里写图片描述
思路:对于每个点,处理出来这个点向左最远能扩展多远,向右最远能扩展多远。处理完后扫一遍就是结果。
写了一发用前缀和预处理的,但是错了。想不到怎么修改。。。
参考题解:http://blog.csdn.net/caduca/article/details/48007559
我是想的处理出来最远能够扩展到的位置,然后扫的时候 最后的位置-最前的位置 就是结果。题解中是处理出来能扩展的长度,扫的时候两边加上就是结果了。
处理左半边的字符串:
‘1’表示1,’0’表示-1,处理前缀和,加和为cur,当cur<0的时候,表示0比1多,当前的长度就是最大长度,当cur >= 0的时候,如果cur+1这个数值没有出现过,则向左扩展最大长度为0,如果曾经出现过,则两点间的串0比1多,因为有前缀和可知,若两点前缀和相同,都为x,则中间那段加和为0,若存在x+1,则相当于前边那个位置后移了1为,且那一位为1,所以加和是x+1,中间段就少个1,0比1多。

#include <bits/stdc++.h>using namespace std;const int MAXN = 1001000;int num[MAXN];int mark[MAXN];int l[MAXN];int r[MAXN];int getStr(){    int len = 0;    char ch;    while((ch = getchar()) != '\n')    {        if(ch == '1') num[len++] = 1;        else num[len++] = -1;    }    return len;}int main(){    int len = getStr();    memset(mark,-1,sizeof(mark));    int cur = 0;    for(int i = 0; i < len; ++i)    {        cur = cur + num[i];        if(cur < 0)            l[i] = i+1;        else        {            if(mark[cur+1] != -1)                l[i] = i - mark[cur+1];            else            {                mark[cur] = i;                l[i] = 0;            }        }    }    memset(mark,-1,sizeof(mark));    cur = 0;    for(int i = len-1; i >= 0; --i)    {        cur = cur + num[i];        if(cur > 0)            r[i] = len-i;        else        {            if(mark[-(cur-1)] != -1)                r[i] = mark[-(cur-1)] - i;            else            {                mark[-(cur)] = i;                r[i] = 0;            }        }    }    int res = 0;    for(int i = 0; i < len; ++i)    {        if(l[i] > 0 && r[i+1] > 0)            res = max(res,l[i]+r[i+1]);    }    printf("%d\n",res);    return 0;}