在获取第K大元素的时候,首先想到的应该就是如何对数据排序,但是排序总是会出现空间复杂度和时间复杂度。
因为我是分类学习的,所以这个题目分配到了“堆”里面,我就查询“java 堆”,结果出现了“PriorityQueue”,然后我就学习了下这个,具体学习可以参考我的博客:https://blog.csdn.net/ChaoFeiLi/article/details/100123155
然后我就发现
使用了priorityQueue之后,每次添加了一个数据之后,他是默认从小到大重新排序
这个问题是第k个最大值,思路应该是从小到大排序,然后倒数第k个就是第K个最大值。注意:从反方向看数组
我们可以创建一个priorityQueue
- priorityQueue中为空,我们添加元素,并且保持priorityQueue.size()<k :这样就保证priorityQueue.peek(),就是第K大的元素
- 如果将要插入的值val>priorityQueue.peek(),则删除priorityQueue的第一个数,添加val
- 如果将要插入的值val<=priorityQueue.peek(),不操作
- 如何一直重复
public class KthLargest {
private PriorityQueue<Integer> queue;
private int k;
public KthLargest(int k, int[] nums) {
this.k = k;
this.queue = new PriorityQueue<Integer>();
for (int i = 0;i<nums.length;i++){
add(nums[i]);
}
}
public int add(int val) {
// 使用了priorityQueue之后,添加了一个数据之后,他是默认从小到大重新排序的
if(queue.size()<k){
queue.add(val);
}else if (!queue.isEmpty() && queue.peek()<val)
{
queue.poll();
queue.offer(val);
}
return queue.peek();
}
}