题目描述
给定一个非空且只包含非负数的整数数组 nums,数组的度的定义是指数组里任一元素出现频数的最大值。
你的任务是在 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 范围内的整数。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/degree-of-an-array
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
解题思路
核心是使用HashMap,Key是数组中的数字,Value是一个数组 [出现的次数, 第一次出现的位置, 最后一次出现的位置] ,找出其中出现次数最多(可能有多个)的数字中,最后一次出现的位置 减去 第一次出现的位置再加上 1 的最小值,这个寻找过程可以在循环中同步进行,不用再遍历一次HashMap。
代码详解
class Solution {
public int findShortestSubArray(int[] nums) {
HashMap<Integer, int[]> hm = new HashMap<>();
int max_count = 0; // 出现次数的最大值
int min_len = Integer.MAX_VALUE; // 最小长度
for(int i = 0; i < nums.length; ++i) { // 仅用一次for
// 这个if-else是添加/修改HashMap的分支
if(hm.containsKey(nums[i])) { // 如果已存在key
++hm.get(nums[i])[0]; // 出现次数+1
hm.get(nums[i])[2] = i; // 更新最后一次出现的位置
} else { // 如果不存在key
hm.put(nums[i], new int[]{1, i, i}); // 新建一个映射,出现次数为1,第一次出现与最后一次出现的位置均为i
}
// 这个if-else是更新“出现次数最大值”和“最小长度”的分支
if(hm.get(nums[i])[0] > max_count) { // 如果有更多的出现次数
max_count = hm.get(nums[i])[0]; // 更新次数最大值
min_len = hm.get(nums[i])[2] - hm.get(nums[i])[1] + 1; // 直接更新最小长度
} else if(hm.get(nums[i])[0] == max_count) { // 如果出现次数等于最大值
min_len = Math.min(min_len, hm.get(nums[i])[2] - hm.get(nums[i])[1] + 1); // 取二者的最小值以更新最小长度
}
}
return min_len;
}
}
注意点
- 可以像本人一样在循环中就同步寻找出现次数最大值和最小长度,也可以把寻找的过程再单独开一个for循环遍历HashMap。