解题思路:
在原数组中,令男生为 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;
}