LeetCode-697.陣列的度
阿新 • • 發佈:2018-11-24
題目
給定一個非空且只包含非負數的整數陣列 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範圍內的整數。
解題
- 首先找出每個數字出現的次數
- 找出最大的度的數字
- 找出具有最大的度的最小子陣列長度, 程式碼如下:
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;
}
}