题目描述:输入n个整数,找出其中最小的K个数。
思路:先将前K个数放入数组,进行堆排序,若之后的数比它还小,则进行调整
//以最大堆来实现,适合海量数据
//最大堆中,根节点的值总是大于它的子树中任意节点的值
//先将前K个数放入数组,进行堆排序,
// 若之后的数比它还小,则进行调整
package Function;
import java.util.ArrayList;
import java.util.Arrays;
public class GetLeastNumbers_Solution40 {
public ArrayList<Integer> GetLeastNumbers_Solution(int[] input, int k) {
ArrayList<Integer> list = new ArrayList<>();
if (input == null || k <= 0 || k > input.length) {
return list;
}
int[] kArray = Arrays.copyOfRange(input, 0, k);
// 创 建 大 根 堆
buildHeap(kArray);
for (int i = k; i < input.length; i++) {
if (input[i] < kArray[0]) {
kArray[0] = input[i];
maxHeap(kArray, 0);
}
}
for (int i = kArray.length - 1; i >= 0; i--) {
list.add(kArray[i]);
}
return list;
}
public void buildHeap(int[] input) {
for (int i = input.length / 2 - 1; i >= 0; i--) {
maxHeap(input,i);
}
}
private void maxHeap(int[] array, int i) {
int left = 2 * i + 1;
int right = left + 1;
int largest = 0;
}
}