数组中第K大元素
在面试的时候遇到过这个问题,后来在leetcode上也遇到了这个问题,于是记录下来以便以后快速回忆。
题目有多种思路,全排序是比较直观的想法,然而最低的时间复杂度为O(nlgn),并且不符合该题目的初衷。
题目更多想问的是如何在不进行全排序的条件下找到数组中第K大的元素,个人认为比较被面试官中意的解答有两个(理应也有其他的..),如下:
构建最大堆
简单分析时间复杂度: 构建堆的时间O(n),k次取最大元素要k(lgn),最终的时间复杂度为O(n+klgn).
显然是比全排序要好一些,但是当时面试的时候我回答这个思路,面试官认为并不是一个好的思路,因为面试官认为构建最大堆更适合题目:求数组中前K大的元素(维护一个大小为K的最大堆).
然而我还是认为这是一个可以的思路…..恩,代码还是贴一下,以后就省事了..
int heap_size;
int parent(int idx) {
return (idx - 1) >> 1; //求父节点
}
int left(int idx) {
return (idx << 1) + 1;//左子树
}
int right(int idx) {
return (idx << 1</