[计蒜客]T1853 非常男女

解题思路：

在原数组中，令男生为 1，女生为 −1 ，这样只要找到两个位置的前缀和相等，他们之间就是男女相等的连续子序列。我们记录每个前缀和第一次出现的位置，下次出现的时候，减去第一次的位置就是长度

为什么?

设前缀和数组为sums[],当sums[right] == sum[left]时，说明[left+1,right]是一个男女数相等的区间,这个区间的长度是right - (left + 1) +1 = right - left。

代码：

#include<iostream>
#include<vector>
#include<unordered_map>
using namespace std;

int main()
{
	int n;
	cin>>n;
	vector<int>nums(n + 1,0);
	unordered_map<int,int>hash;
	hash[0] = 0;//防止 1 0 0 1这种
	int ans = 0;
	for(int i = 1; i <= n; i++)
	{
		cin>>nums[i];
		if(nums[i] == 0)
		{
			nums[i] = -1;
		} 
		nums[i] += nums[i-1];
		if(hash.find(nums[i]) == hash.end())
		{//第一次出现
			hash[nums[i]] = i;
		}
		else if(i - hash[nums[i]] > 0 && (i - hash[nums[i]] > ans))
		{//出现过
			ans = i - hash[nums[i]];
		}
	} 
	cout<<ans;
	return 0;
}