LeetCode 697. 数组的度
给定一个非空且只包含非负数的整数数组 nums, 数组的度的定义是指数组里任一元素出现频数的最大值。
你的任务是找到与 nums 拥有相同大小的度的最短连续子数组,返回其长度。
示例
输入: [1, 2, 2, 3, 1]
输出: 2
解释:
输入数组的度是2,因为元素1和2的出现频数最大,均为2.
连续子数组里面拥有相同度的有如下所示:
[1, 2, 2, 3, 1], [1, 2, 2, 3], [2, 2, 3, 1], [1, 2, 2], [2, 2, 3], [2, 2]
最短连续子数组[2, 2]的长度为2,所以返回2.
用unordered_map<int,vector>,元素做key,元素的下标放入second,就可以知道相同元素的最短距离
注意一下因为这个是未排序的,所以顺序混乱,判断是不能简单写成ans=min(ans,dis)的,只有两个元素度相同时才能这样判断
#include<vector>
#include<iostream>
#include<unordered_map>
#define rep(i,a,b) for(int i=(a); i<(int)(b);++i)
#define _rep(i,a,b) for(int i=(a);i<=(int)(b);++i)
using namespace std;
static const auto io_sync_off = []() {
std::ios::sync_with_stdio(false);
std::cin.tie(nullptr);
return nullptr;
}();
int infer(vector<int>& nums)
{
unordered_map<int,vector<int>> m;
rep(i,0,nums.size())
m[nums[i]].push_back(i);
int deg=0,ans=nums.size();
for(auto it=m.begin();it!=m.end();++it)
{
int s=it->second.size();
if(s>=deg)
{
int dis=it->second[s-1]-it->second[0]+1;
if(s>deg)
{
deg=s;
ans=dis;
}
else
ans=min(ans,dis);
}
}
return ans;
}
int main()
{
int n,num;
cin>>n;
vector<int> a;
rep(i,0,n)
{
cin>>num;
a.push_back(num);
}
cout<<infer(a);
return 0;
}