cf873B-Balanced Substring(前缀和)

来源:互联网 发布:linux显示文本名 编辑:程序博客网 时间:2024/05/22 23:28

题目来源:http:///problemset/problem/873/B

题意

给出以一个01串,其中他的子串只要拥有同样数目的0和1,那么就被称为平衡串。问,给出的串的最长平衡串的长度是多少

思路

起初在本子上写下了三种思路。。(菜。。。)
前缀和,滑动窗,动态规划
然后前缀和这个思路只是简单的把0变成-1之后我就没再看见思路。。。(好菜)
接下来的两种自然也是没什么方法(太弱了。。。)
最后,临近上课,忽然发现之前写的把0变成-1的串好像有规律。。。。
哇,,,好菜。。。
比如前n项和从1变成1,那么两个1之间的不就是平衡串么,。。。
菜的一匹。。。
有一种是特殊的:
111000
1 1 1 -1 -1 -1
前缀和是
1 2 3 2 1 0
那么这个0之前没有出现过,但是想到起初的值就是0。。。所以就特判了下。。

代码

#include<map>#include<cstdio>#include<cstring>#include<algorithm>using namespace std;char s[100000+10];int arr[100000+10];map<int,int> mp;int main(){        int n;        scanf("%d",&n);        scanf("%s",s+1);        for(int i=1; i<=n; i++)        {            arr[i]=s[i]=='0'?-1:1;        }        mp.clear();        int tmp=0,maxx=0;        for(int i=1; i<=n; i++)        {            tmp+=arr[i];            if(tmp!=0&&!mp[tmp])            {                mp[tmp]=i;            }            else            {                if(i-mp[tmp]>maxx)                    maxx=i-mp[tmp];            }        }        printf("%d\n",maxx);}