leetcode算法题每日一练-数组的度

题目

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

分析

题目本身并不复杂，只要理解数组度的概念即可迅速得到答案。根据题目给的解释，数组的度即为数组中元素最大的重复次数。如示例1

[1, 2, 2, 3, 1]
度为2，因为1和2分别重复两次 

同理，示例二

[1,2,2,3,1,4,2]
度为3，因为只有2是重复次数最多的，为3次。

既然明白了度的概念，求最短子数组就很轻松了。首先最短子数组必须满足度的概念，即包含所有重复的元素。如示例2 ，度为三，最短子数组即为

2,2,3,1,4,2

也就是拿重复元素的最大下标减去最小下标。还是示例2

[1,2,2,3,1,4,2]
最短子数组长度为2的最大数组下标即nums[6]
减去最小下标即nums[1]，所以最短子数组为6 - 1 + 1=6 
之所以需要+1是因为数组下标从0开始计数。

实现

 public static  int findShortestSubArray(int[] nums) {
        // key为数组元素，value为存储元素在数组中的下标list
        Map<Integer, List<Integer>> countMap = new HashMap<>();
        for (int i = 0; i < nums.length; i++) {
            // map存在该元素，说明该元素重复，就记录元素的下标
            if (countMap.containsKey(nums[i])){
                List<Integer> list = countMap.get(nums[i]);
                list.add(i);
                countMap.put(nums[i],list);
            }else {
                // 不存在就添加
                List<Integer> integers = new LinkedList<>();
                integers.add(i);
                countMap.put(nums[i], integers);
            }
        }
        // 记录最大的度
        AtomicInteger count = new AtomicInteger(1);
        // 记录对应最大的度的最短子数组
        AtomicInteger min = new AtomicInteger(nums.length);
        // 获取最大的度
        countMap.forEach( (k, v )->{
            if (v.size() >= count.get()){
                count.set(v.size());
            }
        });
        // 计算度最大的最短子数组
        countMap.forEach( (k, v )->{
            if (v.size() == count.get()){
                min.set(Math.min(v.get(v.size() - 1) - v.get(0) + 1, min.get()));
            }
        });

        return min.get();
    }

总结

运行结果需要一个map去存储数组的长度所以空间复杂度为O(n),时间复杂度只进行了数组的遍历并未进行嵌套循环所以为O(n).附上官方地址数组的度