题解 | #最小的K个数#

最小的K个数

https://www.nowcoder.com/practice/6a296eb82cf844ca8539b57c23e6e9bf

import java.util.*;


public class Solution {
    /**
     * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
     *
     * 
     * @param input int整型一维数组 
     * @param k int整型 
     * @return int整型ArrayList
     */
    public ArrayList<Integer> GetLeastNumbers_Solution (int[] input, int k) {
        ArrayList<Integer> res = new ArrayList<>();
        if(input.length == 0||k == 0)
        return res;
       
       int[] sortArr = new int[k];
        for(int i = 0;i < k;i++){
            sortArr[i]=input[i];
        }
         //剪枝-保证前k有序
         Arrays.sort(sortArr);
         for(int i = k;i < input.length;i++){
            if(sortArr[k-1] > input[i]){
                int temp = input[i];
                input[i] = sortArr[k-1];
                sortArr[k-1] = temp;
                Arrays.sort(sortArr);
            }
         }
         for(int i = 0;i<k;i++){
            res.add(sortArr[i]);
         }
        return res;
    }
  
}

全部评论

相关推荐

点赞 评论 收藏
转发
点赞 收藏 评论
分享
牛客网
牛客企业服务