题目
在未排序的数组中找到第 k 个最大的元素。请注意,你需要找的是数组排序后的第 k 个最大的元素,而不是第 k 个不同的元素。
示例
输入: [3,2,1,5,6,4] 和 k = 2
输出: 5
输入: [3,2,3,1,2,4,5,5,6] 和 k = 4
输出: 4
代码,千万别sort一下,然后从后往前找第k大的数,这样这道题目就没意义了
class Solution {
public int findKthLargest(int[] nums, int k) {
//第一种方法,使用优先级队列,默认构建的是我们熟悉的小顶堆
PriorityQueue<Integer>queue=new PriorityQueue<>();
for(int i=0;i<nums.length;i++){
queue.add(nums[i]);
}
int index=nums.length-k;
while(index>0){
index--;
queue.poll();
}
return queue.peek();
}
}
class Solution {
public int findKthLargest(int[] nums, int k) {
int max=nums[0];
int min=nums[0];
//计算数组中的最小值和最大值,依据最小值和最大值构建一个数组
//将nums中的所有元素存入数组中
for(int i=1;i<nums.length;i++){
max=Math.max(max,nums[i]);
min=Math.min(min,nums[i]);
}
//虽然这种方法的时间复杂度很低
//但是空间复杂度却很高,例如假设nums中最大值为Integer.MAX_VALUE
//最小值为0,那么你构建数组的长度为Intege.MAX_VALUE+1
int[]num=new int[max-min+1];
for(int i=0;i<nums.length;i++){
num[nums[i]-min]++;
}
for(int i=num.length-1;i>=0;i--){
if(num[i]>0){k-=num[i];}
if(k<=0){
return i+min;
}
}
return 0;
}
}