最小的K个数（堆排序实现）-Java

题目链接在这里

描述

给定一个数组，找出其中最小的K个数。例如数组元素是4,5,1,6,2,7,3,8这8个数字，则最小的4个数字是1,2,3,4。
0 <= k <= input.length <= 10000
0 <= input[i] <= 10000

这道题很多方法可以做，比如：

• 冒泡排序，执行k次。
• 选择排序，执行k次。
• 堆排序。

下面分别来实现以下。

冒牌版：

需要注意边界

import java.util.ArrayList;

public class Solution {
    public ArrayList<Integer> GetLeastNumbers_Solution(int [] input, int k) {
        ArrayList<Integer> al = new ArrayList<Integer>();
        
        if(input.length < k  || k == 0){
            return al;
        }
        
        for(int i = 0; i < k; i++){
            for(int j = 1; j < input.length; j++){
                if(input[j] > input[j - 1]){
                    int temp = input[j - 1];
                    input[j - 1] = input[j];
                    input[j] = temp;
                }
            }
            al.add(input[input.length - i - 1]);
        }
        return al;
    }
}

堆排序版

堆排序的逻辑比较抽象，画图很麻烦，看这位大神画的吧：传送门

import java.util.ArrayList;

public class Solution {
    public ArrayList<Integer> GetLeastNumbers_Solution(int [] input, int k) {
        ArrayList<Integer> al = new ArrayList<Integer>();
        // 边界检查
        if(input.length < k  || k == 0){
            return al;
        }
		// 排序
        sort(input);
        // 取前k个
        for(int i = 0; i < k; i++){
            al.add(input[i]);
        }
        return al;
    }
    public void sort(int[] input){
        // 构建大顶堆，从第一个非叶子结点开始。
        for(int k = input.length/2 - 1; k >= 0; k--){
            stackSort(input, k, input.length);
        }
        //堆顶元素和堆尾的元素交换，并重新调整堆
        for(int j = input.length - 1; j > 0; j--){
            swap(input, 0 , j);
            stackSort(input, 0 , j);
        }
        
    }
    //调整堆
    public void stackSort(int[] input, int i, int length){
        int temp = input[i];
        /*i 是第一个非叶子结点，k=2i+1表示i的子节点。
        循环条件是k = 2k+1，第一个非叶子结点不会走这个逻辑，
         但第二个、第三个等非叶子结点调整完后，它的子树也需要重新调整。
        */
        for(int k = 2*i+1; k < length; k = 2*k+1){ 
        
            // 注意需要判断k+1是否小于length
            if(k+1<length && input[k] < input[k+1]){  //i是父节点，k和k+1是子节点，判断k和k+1节点的值哪个大。 
                k++;
            }
            //判断子节点中的最大值  与 i节点取值的大小关系。因为i会变，因此提前赋值给temp。
            if(input[k] > temp){
                input[i] = input[k];
                i = k;
            }else{
                break;
            }
        }
        // 赋值到合适的位置
        input[i] = temp;
    }
    public void swap(int[] input, int i, int j){
        int temp = input[i];
        input[i] = input[j];
        input[j] = temp;
    }
}