题目链接
题目简介
给定一个非空且只包含非负数的整数数组 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.
题目解析
- 第一眼看到题目,懵~~~这是个啥???
- 给你一个数组,让你求出来这个数组的众数,众数就叫做度
- 在原本的数组中,任意的连续子数组,找出符合这个度的并且子数组的长度是最小的
- 第一种做法(O(N^2)):
- 利用 HashMap 求出众数
- 遍历一遍 HashMap,求出
出现次数等于众数的数字
,放入 HashSet - 利用双指针,找出该数字在数组中
最开始出现的地方(left)
和最后出现的地方(right)
,并right - left + 1
- 求出最小的 min 即可
- 第二种做法(官方做法-O(N))
- HashMap 定义成
HashMap map
,对于数组来说,int[0]:当前元素出现的次数
int[1]:当前元素第一次出现的地方
int[2]:当前元素最后一次出现的地方
- 找出出现最多的
众数
,求出最小的int[2] - int[1]
- 思路和第一种方法差不多,优点在于:
在进行第一遍遍历的时候,将数组的初始、末尾位置记录下来
题目代码
个人做法
class Solution { public int findShortestSubArray(int[] nums) { // 数组的度:数组里任一元素出现频数的最大值 // 第一步:先求众数 int zongshu = 1; Map<Integer,Integer> map = new HashMap<>(); for(int i = 0; i < nums.length; i++){ if(map.containsKey(nums[i])){ int x = map.get(nums[i]); map.put(nums[i],++x); zongshu = Math.max(zongshu, x); }else{ map.put(nums[i], 1); } } // 第二步:找数组中 最短连续子数组 // 我们已经得到了众数,根据条件可以得出,我们要求的子数组的长度始终是 最后一次出现和第一次出现的距离 // 遍历当前map == 众数的,然后求出 最小 HashSet<Integer> set = new HashSet(); for(int i = 0; i < nums.length; i++){ if(map.get(nums[i]) == zongshu){ set.add(nums[i]); } } int min = Integer.MAX_VALUE; for(Integer i : set){ min = Math.min(min, findShortestSubArray2(nums, i)); } return min; } public int findShortestSubArray2(int[] nums, int x){ int left = 0; int right = 0; int i = 0; for(i = 0; i < nums.length; i++){ if(x == nums[i]){ left = i; break; } } for(i = i; i < nums.length; i++){ if(x == nums[i]){ right = i; } } return right - left + 1; } }
官方做法
class Solution { public int findShortestSubArray(int[] nums) { Map<Integer, int[]> map = new HashMap<Integer, int[]>(); int n = nums.length; for (int i = 0; i < n; i++) { if (map.containsKey(nums[i])) { map.get(nums[i])[0]++; map.get(nums[i])[2] = i; } else { map.put(nums[i], new int[]{1, i, i}); } } int maxNum = 0, minLen = 0; for (Map.Entry<Integer, int[]> entry : map.entrySet()) { int[] arr = entry.getValue(); if (maxNum < arr[0]) { maxNum = arr[0]; minLen = arr[2] - arr[1] + 1; } else if (maxNum == arr[0]) { if (minLen > arr[2] - arr[1] + 1) { minLen = arr[2] - arr[1] + 1; } } } return minLen; } }