剑指offer--最小的k个数

题目描述:输入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;
    }
}

你可能感兴趣的:(剑指offer)