来自学长的快乐AK题——Day2 D

D——差分。。。之类的东东

Description

题面有误，应该为子串长度。（否则这题就sb到爆了）

思路

考虑差分的思想，为男生则赋值+1，为女生则赋值-1，做前缀和。

倘若一段区间i~j内男女个数相等，则该区间的所有值之和显然等于0（因为个数相等的+1和-1互相抵消了）。
用前缀和来表示，即s[j]-s[i-1]=0。简单移项可以得到s[j]=s[i-1]。

也就是说，如果s[j]=s[i]，那么i+1~j就是一个男女人数相等的区间。用桶记录之前出现过的最早的s数组值的位置（因为要求最大），倘若之后再有相同的值，用当前位置减去之前位置再与答案比较即可。

代码

#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
const int N=1e6;
int n,s[N],h[N],ans; 
int main()
{
	scanf("%d",&n);
	for(int i=1,k;i<=n;i++)
	{
		scanf("%d",&k);
		if(k==0) k=-1;
		s[i]=s[i-1]+k;
	}
	for(int i=1;i<=n;i++)
	{
		int t=s[i]+n;
		if(h[t]==0) h[t]=i;
		else ans=max(ans,i-h[t]);
	}
	printf("%d",ans);
	return 0;
}