解題思路:
在原數組中,令男生爲 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;
}