3、leetcode 697 数组的度

leetcode 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.

示例 2：
输入：[1,2,2,3,1,4,2]
输出：6

提示：
nums.length 在1到 50,000 区间范围内。
nums[i] 是一个在 0 到 49,999 范围内的整数。

一、利用HashMap

class Solution{
    public int findShortestSubArray(int[] nums){
        //int[]中存放三个元素,nums[i]的start索引,nums[i]的end索引以及num[i]的数量count
        HashMap<Integer,int[]> map=new HashMap<>();
        int count=0;
        int res=1;
        for(int i=0;i<nums.length;i++){
            if(map.containsKey(nums[i])){
                int[] arr=map.get(nums[i]);
                arr[1]=i;//arr[1]存放的是nums[i]最后一次出现的索引
                arr[2]+=1;//arr[2]存放的是数量
                if(count<arr[2]){
                    count=arr[2];
                    res=arr[1]-arr[0]+1;
                }else if(count==arr[2]){
                    res=Math.min(res,arr[1]-arr[0]+1);
                }
            }else{
            	//元素第一次出现，进行初始化
                int[] arr=new int[3];
                arr[0]=i;
                arr[1]=i;
                arr[2]=1;
                map.put(nums[i],arr);
            }
        }
        return res;

    }
}

一、计数排序思想+滑动窗口

滑动窗口模板见滑动窗口模板Java版

class Solution{
    public int findShortestSubArray(int[] nums){
        int[] count=new int[50000];
        //计数排序思想
        for(int n:nums){
            count[n]++;
        }
        //记录数组的度
        int degree=count[0];
        for(int n:count){
            degree=Math.max(degree,n);
        }
        count=new int[50000];
        int left=0;
        int right=0;
        int res=Integer.MAX_VALUE;
        //滑动窗口思想
        while(right<nums.length){
            count[nums[right]]++;
            while(degree==count[nums[right]]){
                res=Math.min(res,right-left+1);
                count[nums[left]]--;
                left++;
            }   
            right++;
        }
        return res;
    }
}