题目大意:
题目链接:
洛谷:https://www.luogu.org/problemnew/show/P1114
JZOJ:https://jzoj.net/senior/#main/show/3913
给出长度为 n n n的01串,求其中长度最长且0的数量等于1的数量的子串长度。
思路:
作为本次比赛最简单的题目,我居然没做出来,还是太菜了QWQ。
用
1
1
1表示男生,
−
1
-1
−1表示女生,
s
i
s_i
si表示前缀和,那么当两个
s
i
s_i
si相等时,这一串就是一个合法的子串。
那么就记录
s
i
s_i
si出现的最早的时候,每次再寻找到
s
i
s_i
si就直接更新答案。
本来时间复杂度是
O
(
n
)
O(n)
O(n)的,但是有负数,懒得加
n
n
n,用
m
a
p
map
map水了个
O
(
n
l
o
g
n
)
O(n\ logn)
O(n logn),也是可以过的。
代码:
#include <cstdio>
#include <map>
using namespace std;
const int N=100010;
int n,s[N],ans;
map<int,int> t;
int main()
{
scanf("%d",&n);
for (int i=-N;i<=N;i++)
t[i]=-1; //初始化map
t[0]=0;
for (int i=1;i<=n;i++)
{
scanf("%d",&s[i]);
if (!s[i]) s[i]=-1;
s[i]+=s[i-1];
if (t[s[i]]>=0) ans=max(ans,i-t[s[i]]);
else t[s[i]]=i;
}
printf("%d\n",ans);
return 0;
}