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

版权声明：若希望转载，在评论里直接说明即可，谢谢！ https://blog.csdn.net/SSL_ZYC/article/details/86628396

题目大意：

题目链接：

洛谷：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(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;
}