[計蒜客]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;
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章