题目描述
对于一个无序数组,数组中元素为互不相同的整数,请返回其中最小的k个数,顺序与原数组中元素顺序一致。
给定一个整数数组A及它的大小n,同时给定k,请返回其中最小的k个数。
测试样例
[1, 2, 3, 4], 4, 2
返回:[1, 2]
解析
import java.util.*;
public class KthNumbers {
public int[] findKthNumbers(int[] A, int n, int k) {
// write code here
if (n < k) return null;
if (k == n) return A;
int[] b = Arrays.copyOf(A, n);
Arrays.sort(b);
int target = b[k - 1];
int[] arr = new int[k];
int p = 0;
for (int i = 0; i < n; i++) {
if (A[i] <= target) {
arr[p++] = A[i];
}
}
return arr;
}
}