开启掘金成长之旅！这是我参与「掘金日新计划 · 12 月更文挑战」的第31天，点击查看活动详情
一、题目描述：
给定一个非空且只包含非负数的整数数组 nums，数组的 度 的定义是指数组里任一元素出现频数的最大值。
你的任务是在 nums 中找到与 nums 拥有相同大小的度的最短连续子数组，返回其长度。
示例 1：
输入：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 。
示例 2：
输入：nums = [1,2,2,3,1,4,2]
输出：6
解释：
数组的度是 3 ，因为元素 2 重复出现 3 次。
所以 [2,2,3,1,4,2] 是最短子数组，因此返回 6 。
提示：
- nums.length 在 1 到 50,000 范围内。
- nums[i] 是一个在 0 到 49,999 范围内的整数。
二、思路分析：
- 由于nums[i] 是一个在 0 到 49,999 范围内的整数，因此使用int数组代替哈希表；
- 考虑数组只有两个元素的特例，相等则为2，反之则为1；
- 计算数组的度；
- 考虑度为1的特例，直接返回1；
- 计算最短连续子数组，找到其最小索引i，并以i+1找到其最大索引j，长度为j-1+1;
三、AC 代码：
class Solution {
public:
int findShortestSubArray(vector<int>& nums) {
// 两个元素的特例
if(nums.size() == 2) {
return nums[0] == nums[1] ? 2 : 1;
}
// 构造哈希表
int array[50000] = {0};
// 计算度
int degree = 0;
for(auto num : nums) {
++array[num];
degree = array[num] > degree ? array[num] : degree;
}
// 度为1的特例
if(degree == 1) return 1;
// 计算最短子数组
int res = 0;
for(int i=0; i<nums.size(); ++i) {
if(array[nums[i]] == degree) {
// 此时i为最短连续子数组的最小索引
int cur_degree = degree;
for(int j=i+1; j<nums.size(); ++j) {
if(nums[j] == nums[i]) {
--cur_degree;
}
if(cur_degree == 1) {
// 此时j为最短连续子数组的最大索引
int cur_diff = j - i + 1;
if(res == 0) {
res = cur_diff;
} else {
res = cur_diff < res ? cur_diff : res;
}
// 哈希表相应元素清零，避免重复计算
array[nums[i]] = 0;
break;
}
}
}
}
return res;
}
};
