LeetCode 697. 数组的度

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;
}