題目描述

輸入n個整數，找出其中最小的K個數。例如輸入4,5,1,6,2,7,3,8這8個數字，則最小的4個數字是1,2,3,4,。

分析：

找出最小的K個數字，可以通過維護一個大根堆來實現最小的K個數字，當大根堆的個數大于K的時候，把堆頂元素去掉，從而保證大根堆的大小為k。
同樣，也可以通過使用冒泡排序算法，每次排序得到最小的數字的一個位置。這樣就可以實現不用全部排序也可以得到最小的四個數字。

又是一道簡單題，就不進行詳細的分解了。

解答：

import java.util.ArrayList;
public class Solution {
    public ArrayList<Integer> GetLeastNumbers_Solution(int [] input, int k) {
        ArrayList<Integer> leastk = new ArrayList<Integer>();
        if(input==null || input.length==0 || input.length<k) return leastk;
        for(int i =0;i<k;i++) {
            //冒泡排序，每次排序過程確定一個最小值
            for(int j = input.length-1;j>i;j--){
                if (input[j - 1] > input[j]) { //交換，把小的頂到前面去
                    int temp = input[j];
                    input[j] = input[j - 1];
                    input[j - 1]= temp;
                }
            }
            leastk.add(input[i]);
        }
        return leastk;
    }
}

運行結果