数组中的第K个最大元素

Posted lijianming180

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了数组中的第K个最大元素相关的知识,希望对你有一定的参考价值。

Contents

题目

在未排序的数组中找到第 k 个最大的元素。请注意,你需要找的是数组排序后的第 k 个最大的元素,而不是第 k 个不同的元素。
示例 1:
输入: [3,2,1,5,6,4] 和 k = 2
输出: 5
示例 2:
输入: [3,2,3,1,2,4,5,5,6] 和 k = 4
输出: 4
说明:
你可以假设 k 总是有效的,且 1 ≤ k ≤ 数组的长度。

思路1

插入一个k大小的最小堆,
满了之后,再最小值比较,然后从上到下堆化

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
class Solution {
public int findKthLargest(int[] nums, int k) {
Heap heap = new Heap(k);
for(int i = 0; i< nums.length ; i++){
heap.insert(nums[i]);
}
return heap.maxK();
}
static class Heap{
int[] nums;
int count;
int capacity;
public Heap(int capacity){
this.nums = new int[capacity+1];
this.count = 0;
this.capacity = capacity;
}

public void insert(int num){
if( count < capacity){
notFullInsert(num);
}else{
fullInsert(num);
}
}
public void fullInsert(int num){
int min = nums[1];
if(num < min){
return;
}
nums[1] = num;
int minPos = 1;
int i = 1;
while(true){
if(i*2 <= capacity && nums[i] > nums[i*2]){minPos = i*2;}
if(i*2+1 <= capacity && nums[minPos] > nums[i*2+1]){minPos = i*2+1;}
if(i== minPos)break;
swap(nums,i,minPos);
i = minPos;
}

}
大专栏  数组中的第K个最大元素public void notFullInsert(int num){
int index = count+1;
nums[index] = num;
count++;
while(index/2 > 0 && nums[index/2] > nums[index]){
swap(nums,index , index/2);
index = index/2;
}
}
public void swap(int[] array, int fromIndex ,int toIndex){
int tmp = array[toIndex];
array[toIndex] = array[fromIndex];
array[fromIndex] = tmp;
}
public int maxK(){
return nums[1];
}
}
}

思路2

先填充 k个大小的数组,然后从下到上进行堆化,
满了之后,再最小值比较,然后从上到下堆化

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
class Solution {
public int findKthLargest(int[] nums, int k) {
int[] arrays = new int[k+1];

for(int i = 1 ; i<= k ; i++ ){
arrays[i] = nums[i-1];
}
initTree(arrays);

for(int i = k ; i< nums.length ; i++){
int min = arrays[1];
if(min < nums[i]){
arrays[1] = nums[i];
heapify(arrays,1, k);

}
}
return arrays[1];
}
private void initTree(int[] array){
for(int i = array.length/2; i>= 1;i--){
heapify(array, i, array.length -1);
}
}
private void heapify(int[] array,int i, int n){
while(true){
int minPos = i;
if(i*2 <= n && array[i] > array[i*2]){
minPos = i*2;
}
if(i*2+1 <= n && array[minPos] > array[i*2+1]){minPos = i*2+1;}
if(minPos == i)break;
swap(array, i, minPos);
i = minPos;
}
}

public void swap(int[] nums , int fromIndex , int toIndex){
int tmp = nums[toIndex];
nums[toIndex] = nums[fromIndex];
nums[fromIndex] = tmp;
}

以上是关于数组中的第K个最大元素的主要内容,如果未能解决你的问题,请参考以下文章

数组中的第K个最大元素 优先队列

LeetCode.215-数组中的第K个最大元素

Leetcode 215. 数组中的第K个最大元素 By Python

215. 数组中的第K个最大元素

#yyds干货盘点# LeetCode 腾讯精选练习 50 题:数组中的第K个最大元素

LeetCode:数组中的第K个最大元素215