找最小的K个数

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

方法1：利用快速排序的思想，每次取得划分元素的位置partitionIndex。该位置代表key值是整个数组的第partitionIndex+1小。将partitionIndex+1与k比较，如果小于k，继续往右划分。如果大于k，继续往左划分，直到partitionIndex+1与k相等。

方法2： 冒泡排序的思想，最外层只需要循环K次，就可以得到最小的k个数，不用全部排序。

方法3：采用最大堆的思想，初始化最小k个元素的集合为前k个数。后面，将输入数组k+1位置的值与集合中最大的元素作比较。如果小于集合中最大的元素Y，说明最小k个元素的集合应该去掉Y，添加input[k+1]。如果input[k+1]大于集合中最大的元素Y，则集合不变。以此类推比较后面的值和更新过后集合中最大的元素。。。

package alex.test;

import java.util.ArrayList;
import java.util.Collections;
import java.util.Scanner;

public class FindMinKNum {

    public static void main(String[] args) {
        // TODO Auto-generated method stub
        Scanner scanner = new Scanner(System.in);
        System.out.println("input the size of array:");
        int n = scanner.nextInt();
        System.out.println("input the value of array:");
        int[] array = new int[n];
        for (int i = 0; i < n; i++) {
            array[i] = scanner.nextInt();
        }
        System.out.println("input K:");
        int k = scanner.nextInt();
        ArrayList<Integer> result = GetLeastNumbers_Solution2(array, k);
        for (int i = 0; i < result.size(); i++) {
            System.out.println(result.get(i));
        }
    }
    //方法1
    public static ArrayList<Integer> GetLeastNumbers_Solution(int[] input, int k) {
        ArrayList<Integer> list = new ArrayList<Integer>();
        int len = input.length;
        if (len < k || k <= 0) {
            return list;
        }
        int partitionIndex = partition(input, 0, len - 1);
        while (partitionIndex + 1 != k) {
            if (partitionIndex + 1 < k) {
                partitionIndex = partition(input, partitionIndex + 1, len - 1);
            } else {
                partitionIndex = partition(input, 0, partitionIndex - 1);
            }
        }
        for (int i = 0; i < k; i++) {
            list.add(input[i]);
        }
        return list;
    }

    public static int partition(int a[], int low, int high) {
        int key = a[low]; // 划分元素
        while (low < high) {
            while (low < high && a[high] >= key) {
                high--;
            }
            if (low < high) {
                a[low++] = a[high];
            }
            while (low < high && a[low] < key) {
                low++;
            }
            if (low < high) {
                a[high--] = a[low];
            }
        }
        a[low] = key;
        return low;
    }

    // 方法2：冒泡排序的思想，最外层循环K次，不用全部排序。
    public static ArrayList<Integer> GetLeastNumbers_Solution1(int[] input, int k) {
        ArrayList<Integer> list = new ArrayList<Integer>();
        int len = input.length;
        if (k > len) {
            return list;
        }
        for (int i = 0; i < k; i++) {
            for (int j = 0; j < len - i - 1; j++) {
                if (input[j] < input[j + 1]) {
                    int temp = input[j];
                    input[j] = input[j + 1];
                    input[j + 1] = temp;
                }
            }
            list.add(input[len - i - 1]);
        }
        return list;
    }

    // 方法3：堆排序的思想
    public static ArrayList<Integer> GetLeastNumbers_Solution2(int[] input, int k) {
        int len = input.length;
        ArrayList<Integer> list = new ArrayList<Integer>();
        if (k > len) {
            return list;
        }
        for (int i = 0; i < k; i++) {
            list.add(input[i]);
        }
        for (int i = k; i < len; i++) {
            Collections.sort(list);
            int maxOfList = list.get(k - 1);
            if (input[i] < maxOfList) {
                list.remove(k - 1);
                list.add(k - 1, input[i]);
            }
        }
        return list;
    }
}