牛客刷题之最小的k个数

题目描述

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

解题思路：

        采用选择排序的思想即可，注意k的范围。

import java.util.ArrayList;
public class Solution {
     int len = 0;
    public ArrayList<Integer> GetLeastNumbers_Solution(int [] input, int k) {
       ArrayList<Integer> array = new ArrayList<>();
        if (input == null || input.length == 0) {
            return array;
        }
        int len = input.length;
        for (int i = 0; i < len; i++) {
            int min_number = i;
            for (int j = i + 1; j < len; j++) {
                if (input[min_number] > input[j]) {
                    int temp = input[min_number];
                    input[min_number] = input[j];
                    input[j] = temp;
                }
            }
        }
        if (k <= input.length) {
            for (int i = 0; i < k; i++) {
                array.add(input[i]);
            }
        }
        return array;
    } 
    
}