LeetCode刷题: 数组中的第K个最大元素(day40)
题目描述:
在未排序的数组中找到第 k 个最大的元素。请注意,你需要找的是数组排序后的第 k 个最大的元素,而不是第 k 个不同的元素。
示例 1:
输入: [3,2,1,5,6,4] 和 k = 2
输出: 5
示例 2:
输入: [3,2,3,1,2,4,5,5,6] 和 k = 4
输出: 4
说明:
你可以假设 k 总是有效的,且 1 ≤ k ≤ 数组的长度。
思路:桶排序,非常好理解,先遍历一遍数组找出最大最小值。
创建一个桶,长度为max-min+1,桶的索引对应于与Min的差值,桶中装的元素为该值出现次数。
上代码:
class Solution {
public int findKthLargest(int[] nums, int k) {
int max=Integer.MIN_VALUE;
int min=Integer.MAX_VALUE;
for(int num:nums){
max=Math.max(max,num);
min=Math.min(min,num);
}
int[] bucket=new int [max-min+1];
for(int num:nums){
bucket[num-min]++;
}
int count=0;
for(int i=bucket.length-1;i>=0;i--){
count+=bucket[i];
if(count>=k) return min+i;
}
return -1;
}
}
执行结果:
执行耗时: