题目
在未排序的数组中找到第 k
个最大的元素。请注意,它是数组有序排列后的第 k
个最大元素,而不是第 k
个不同元素。
例如,
给出 [3,2,1,5,6,4]
和k = 2
,返回 5
。
注意事项:
你可以假设 k
总是有效的,1 ≤ k ≤
数组的长度。
方法一:
先对数组快速排序再选择第K大的。时间复杂度O(nlgn) 空间复杂度O(1)
方法二:
维持一个大小为K的最小堆,遍历整个数组和堆顶比较。时间复杂度O(n lgK)空间复杂度 O(k)
public int findKthLargest(int[] nums, int k) { //找到第k大的数
PriorityQueue<Integer> heap =new PriorityQueue<>();
for (int i = 0; i < k; i++) {
heap.add(nums[i]);
}
for (int i = k; i < nums.length; i++) {
if (heap.peek()<nums[i]) {
heap.poll();
heap.add(nums[i]);
}
}
return heap.peek();
}
方法三:
类似快排的思想:选择一个数,将比它小的移动到左边,比它大的移动到右边。如果位置是n-k,输出;否则左边或右边继续查找。时间复杂度O(n lgn) 空间复杂度O(1)
方法四:
如果变态要求原数组不能修改(不能移动),并且额外空间复杂度为O(1)
那么上述方法就全都不能用了。考虑用二进制解法。比如对于int型正整数,4字节,即32位,如果第i位 置1,保持前i-1位不变,那么根据前i位可以将数组分成两部分,即可以起到二分的作用。具体举例如下,数组为[2,6,4,1,3,5],然后找第2大的数,即k=2。int型对应32位bit标号为31,30,29...3,2,1,0:
十进制 | 二进制 |
---|---|
2 | 10 |
6 | 110 |
4 | 100 |
1 | 1 |
3 | 11 |
5 | 101 |
1、bit位第2位为1,可以将数组分为[2,1,3]和[6,4,5]两部分。由于需要找第2大,所以在数组[6,4,5]中继续查找。
2、继续考察bit位第1位为1,则数组[6,4,5]可以分为[4,5]和[6]两部分,此时较大的数组为[6]只有一个元素,所以在较小的数组[4,5]中继续找第1大的数(这里要注意一下,由于[6]只有一个,所以k=k-1=1)
3、继续考察第0位为1,则数组[4,5]可以分为[4]和[5]两组,较大数组[5]只有1个,而此时k=1,故可以直接取出5即为所求。
参考: