剑指Offer：最小的K个数（java）

1. 题目描述
输入n个整数，找出其中最小的K个数。例如输入4,5,1,6,2,7,3,8这8个数字，则最小的4个数字是1,2,3,4,。

2. 解题思路
（1）可以先排序数组，然后再把前n个数字放到集合中，返回集合即可。
（2）也可以用堆的思想去做，先将前K个数放入数组，进行堆排序，若之后的数比它还小，则进行调整。

3. 代码实现
（1）

import java.util.ArrayList;

public class Solution {
    public ArrayList<Integer> GetLeastNumbers_Solution(int [] input, int k) {

        ArrayList<Integer> arrayList = new ArrayList<>();
        if(input.length < k || input == null)
            return arrayList;
        int temp = 0;
        for(int i = 0; i < input.length; i++){
            for (int j = i+1; j < input.length; j++){
                if (input[i] > input[j])
                {
                   temp = input[i];
                   input[i] = input[j];
                   input[j] = temp;
                }
            }
            if(i < k)
                arrayList.add(input[i]);
            if (i == k)
                break;
        }
        return arrayList;
    }
}

（2）

import java.util.ArrayList;
import java.util.Arrays;

public class Solution {
    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;
    }

    private 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;
        
        if(left < array.length && array[left] > array[i])
            largest = left;
        else 
            largest = i;
        
        if(right < array.length && array[right] > array[largest])
            largest = right;
        
        if (largest != i){
            int temp = array[i];
            array[i] = array[largest];
            array[largest] = temp;
            maxHeap(array, largest);
        }
    }
}