牛客-NC88-寻找第K大
NC88. 寻找第K大(medium)
方法一:栈法
思路:同牛客-NC119-最小的K个数,求第K大等价于在小根堆中求第N-K+1小,所有改变遍历次数即可:
import java.util.*;
public class Solution {
public int findKth(int[] a, int n, int K) {
// write code here
// 特判
int ret = 0;
if (n == 0 || K > n) {
return ret;
}
PriorityQueue<Integer> pq = new PriorityQueue();
for (int i = 0; i < n; i++) {
pq.offer(a[i]);
}
for (int i = 0; i <= n - K; i++) {
ret = pq.poll();
}
return ret;
}
}
时间复杂度: O(NlogN),建小根堆所需的时间复杂度。
空间复杂度: O(N), 我们使用大小为N的PriorityQueue来存数据。