[牛客网] n个数里最小的k个

找出n个数里最小的k个

• 输入描述：

每个测试输入包含空格分割的n+1个整数，最后一个整数为k值，n不超过100。

• 输出描述：

输出n个整数里最小的k个数，并按升序输出。

• 示例：

输入：
3 9 6 8 -10 7 -11 19 30 12 23 5
输出：
-11 -10 3 6 7

【解题思路】：

快排思想，排序+分区；
找到下标为k的位置，则可知，前k小的都在左边，排序输出即可。

【代码展示】：

import java.util.*;

public class Main {
    
    public static void main(String[] args){
        Scanner sc = new Scanner(System.in);
        
        while(sc.hasNext()){
            String str = sc.nextLine();
            String[] s = str.split(" ");
            int[] num = new int[s.length-1];
            
            for(int i = 0; i < s.length - 1; i++) {//最后一个是k值
                num[i] = Integer.parseInt(s[i]);
            }
            
            int k = Integer.parseInt(s[s.length-1]);//k
            int start = 0;
            int end = num.length-1;
            
            int index = qSort(num, start, end);
            
            while(index != k){
                if(index > k){
                    end = index - 1;
                    index = qSort(num, start, end);
                }else{
                    start = index + 1;
                    index = qSort(num, start, end);
                }
            }
            
            Arrays.sort(num, 0, k);//排序
            
            for(int i=0;i<k;i++){
                System.out.print(i == k-1 ? num[i] : num[i] + " ");//最后一个数后面不再需要空格
            }
        }
    }
   
    public static int qSort(int[] num, int start, int end){
       
        int par = num[start];//快排 找基准
       
        while(start < end){
            while(start < end && num[end] >= par) {//判断start<end不能省
                end--;
            }
            num[start] = num[end];
            
            while(start < end && num[start] < par) {
                start++;
            }
            num[end] = num[start];
        }
        num[start] = par;
        return start;
    }
}