巧妙运用map 力扣 697. 数组的度

力扣 697. 数组的度

给定一个非空且只包含非负数的整数数组 nums, 数组的度的定义是指数组里任一元素出现频数的最大值。

你的任务是找到与 nums 拥有相同大小的度的最短连续子数组，返回其长度。

示例 1:

输入:

[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.

示例 2:

输入:

[1,2,2,3,1,4,2]

输出:

6

注意:

nums.length 在1到50,000区间范围内。 nums[i] 是一个在0到49,999范围内的整数。

思路：
使用<int,vector<int>>类型的 map，整型记录数组中每个数，vector 记录每个数对应位置，vector 的大小即对应数的频率

代码：

class Solution
{
    
    
public:
    int findShortestSubArray(vector<int> &nums)
    {
    
    
        map<int, vector<int>> a;
        for (int i = 0; i < nums.size(); i++)
        {
    
    
            a[nums[i]].push_back(i);
            //记录所有数字对应位置
        }
        int x = 0, ans = INT_MAX;
        for (auto inter = a.begin(); inter != a.end(); inter++)
        {
    
    
            int m = inter->second.size();
            if (x < m)
            {
    
    
                x = m;
                ans = inter->second[m - 1] - inter->second[0] + 1;
                /*这一大堆就是最后一次出现这个数的位置
                和第一次出现这个数的位置做差之后加一，
                得到最短字数串长度*/
            }
            if (x == m)
            {
    
    
                ans = min(ans, inter->second[m - 1] - inter->second[0] + 1);
            }
        }
        return ans;
    }
};