【JZOJ3913】【洛谷P1114】“非常男女”计划【模拟】

题目大意：

题目链接：

洛谷：https://www.luogu.org/problemnew/show/P1114
JZOJ：https://jzoj.net/senior/#main/show/3913

给出长度为$n$的01串，求其中长度最长且0的数量等于1的数量的子串长度。

---

思路：

作为本次比赛最简单的题目，我居然没做出来，还是太菜了QWQ。
用$1$表示男生，$-1$表示女生，$s_i$表示前缀和，那么当两个$s_i$相等时，这一串就是一个合法的子串。
那么就记录$s_i$出现的最早的时候，每次再寻找到$s_i$就直接更新答案。
本来时间复杂度是$O(n)$的，但是有负数，懒得加$n$，用$map$水了个$O(nlogn)$，也是可以过的。

---

代码：

#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;
}