1. 程式人生 > >LeetCode-697.陣列的度

LeetCode-697.陣列的度

題目

給定一個非空且只包含非負數的整數陣列 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範圍內的整數。

解題

  1. 首先找出每個數字出現的次數
  2. 找出最大的度的數字
  3. 找出具有最大的度的最小子陣列長度, 程式碼如下:
class Solution {
    public int findShortestSubArray(int[] nums) {
        Map<Integer, Integer> map = new HashMap<>();
        Set<
Integer>
maxSet = new HashSet<>(); for(int i = 0; i < nums.length; i++){ if(map.containsKey(nums[i])){ map.put(nums[i], map.get(nums[i]) + 1); } else { map.put(nums[i], 1); } } int
maxDegree = 0; for(Map.Entry<Integer, Integer> entry : map.entrySet()){ int value = entry.getValue(); if(value == maxDegree){ maxSet.add(entry.getKey()); } else if (value > maxDegree) { maxDegree = value; maxSet.clear(); maxSet.add(entry.getKey()); } } int minLength = 0; for(int num : maxSet){ int left = 0; int right = nums.length - 1; for(; left < right; left++){ if(nums[left] == num){ break; } } for(; right > left; right--){ if(nums[right] == num){ break; } } if (minLength == 0) { minLength = right - left + 1; } else { minLength = Math.min(right - left + 1, minLength); } } return minLength; } }

執行用時: 96 ms, 在Degree of an Array的Java提交中擊敗了30.73% 的使用者

下面這個是在LeetCode上看到的別人的解法, 只用了8ms, 程式碼如下:

class Solution {
    public static int findShortestSubArray(int[] nums) {

        //求出nums陣列中最大的數
        int maxNum = nums[0];
        for (int i = 1; i < nums.length; i++) {
            maxNum = Math.max(maxNum, nums[i]);
        }

        //用三個陣列分別記錄  起始位置,長度,度
        int max = 1;//最大的度

        int[] start = new int[maxNum + 1];//起始位置
        int[] length = new int[maxNum + 1];//長度
        int[] count = new int[maxNum + 1];//度

        for (int i = 0; i < nums.length; i++) {
            count[nums[i]]++;
            if (count[nums[i]] == 1) {//記錄第一次出現的位置
                start[nums[i]] = i;
            } else {
                length[nums[i]] = i - start[nums[i]] + 1;//記錄當前的長度
                max = Math.max(count[nums[i]], max);//最大的度
            }
        }

        if (max == 1) {
            return 1;
        }

        int degree = nums.length;//最大的度對應的最小的陣列長度
        for (int i = 0; i <= maxNum; i++) {
            if (count[i] == max) {
                degree = Math.min(length[i], degree);
            }
        }

        return degree;
    }

}