小米笔试题:无序数组中最小的k个数

时间:2022-12-09 18:49:25

 题目描述

链接:https://www.nowcoder.com/questionTerminal/ec2575fb877d41c9a33d9bab2694ba47?source=relative
来源:牛客网

无序数组中最小的k个数 对于一个无序数组,数组中元素为互不相同的整数,请返回其中最小的k个数,顺序与原数组中元素顺序一致。 给定一个整数数组A及它的大小n,同时给定k,请返回其中最小的k个数。 测试样例: [1,2,4,3],4,2
返回:[1,2]

 

代码

需要保留K个较小的元素,可以删除n-k个较大的元素。

public class KthNumbers {
    public static int[] findKthNumbers(int[] A, int n, int k) {
        int delNum = n - k;// 要删除的元素数量
        int len = A.length;// 实际数组中的元素个数
        while (delNum > 0) {
            // 找到当前数组中最大的元素
            int maxIndex = 0;// 记录最大元素的下标
            int i;
            for (i = 1; i < len; i  ) {// 寻找最大元素并记录其在数组中的下标
                if (A[maxIndex] < A[i]) {
                    maxIndex = i;
                }
            }
            // 删除当前最大元素即maxIndex
            for (int j = maxIndex; j < len - 1; j  ) {
                A[j] = A[j   1];
            }
            len--;
            delNum--;
        } 

        int result[] = new int[k];
        for (int i = 0; i < k; i  ) {
            result[i] = A[i];
        }
        return result;
    }
}