思路:首先可以将问题抽象成一个括号问题,将0视为-1,1视为1,那么如果0和1的个数相等,他们的前缀和为0,这个时候就可以直接更新答案了,而如果不为0,这个时候如果前面有一段的前缀和等于现在这个前缀和,那么证明两个前缀和之间的和为0,那么更新答案,仔细想想是不是?
#include<bits/stdc++.h>
using namespace std;
const int maxn = 1000000+7;
char s[maxn];
int a[maxn];
int sum[maxn];
map<int,int>pre;
int main()
{
scanf("%s",s+1);
int len = strlen(s+1);
for(int i = 1;i<=len;i++)
if(s[i]=='0')
a[i]=-1;
else a[i]=1;
sum[0]=0;
for(int i = 1;i<=len;i++)
sum[i]=sum[i-1]+a[i];
int ans = 0;
for(int i = 1;i<=len;i++)
{
if(sum[i]==0)
ans = max(ans,i);
else
{
if(pre[sum[i]]==0)
pre[sum[i]]=i;
else ans = max(ans,i-pre[sum[i]]);
}
}
printf("%d\n",ans);
}
基准时间限制:1 秒 空间限制:131072 KB 分值: 20
难度:3级算法题
给定一个0-1串,请找到一个尽可能长的子串,其中包含的0与1的个数相等。
Input
一个字符串,只包含01,长度不超过1000000。
Output
一行一个整数,最长的0与1的个数相等的子串的长度。
Input示例
1011
Output示例
2