【力扣每日一题：2-20】697. 数组的度【简单】

题目链接

697. 数组的度【简单】

题目简介

给定一个非空且只包含非负数的整数数组 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.

题目解析

1. 第一眼看到题目，懵~~~这是个啥？？？
2. 给你一个数组，让你求出来这个数组的众数，众数就叫做度
3. 在原本的数组中，任意的连续子数组，找出符合这个度的并且子数组的长度是最小的
4. 第一种做法(O(N^2))：
   • 利用 HashMap 求出众数
   • 遍历一遍 HashMap，求出出现次数等于众数的数字，放入 HashSet
   • 利用双指针，找出该数字在数组中最开始出现的地方（left）和最后出现的地方(right)，并right - left + 1
   • 求出最小的 min 即可
5. 第二种做法（官方做法-O(N))
   • HashMap 定义成 HashMap<Integer, int[]> 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;
    }
}