输入包含两行,第一行包含两个整数n和k(1≤k≤n≤105),代表数组arr的长度,第二行包含n个整数,代表数组arr。
输出描述:
输出包含一行,k个整数,代表数组中最小的k个整数。
备注:
时间复杂度O(nlog_2k)和O(n)O(nlog2k)和O(n),额外空间复杂度O(n)O(n)
public static void main(String[] args) {
Scanner sc = new Scanner(System.in);
System.out.println("请输入数组长度n和输出最小k个整数,以空格隔开");
String[] num = sc.nextLine().split(" ");
while (num.length <2 || Integer.parseInt(num[0]) < Integer.parseInt(num[1])){
System.out.println("输入格式有误,请重新输入");
num = sc.nextLine().split(" ");
}
int n = Integer.parseInt(num[0]);
int k = Integer.parseInt(num[1]);
System.out.println("请输入数组arr,数字之间以空格隔开");
String[] inputArr = sc.nextLine().split(" ");
int[] numArr = new int[inputArr.length];
for (int i = 0;i< inputArr.length;i++){
numArr[i] = Integer.parseInt(inputArr[i]);
}
// 需要输出最小k个数的数组
int[] outputArr = new int[k];
// 默认取原数组前k个填充输出数组
if (k >= 0) System.arraycopy(numArr, 0, outputArr, 0, k);
// 排序
Arrays.sort(outputArr);
for (int i=k;i< numArr.length;i++) {
// 如果第i个数比输出数组的最后一位小,说明需要放入输出数组
if (numArr[i] < outputArr[k - 1]) {
// 数组后移
change(outputArr,numArr[i]);
}
}
for (int i = 0; i < k; i++) {
System.out.print(outputArr[i] + " ");
}
}
public static void change(int []a,int x){
int i=0;
for (;i<a.length;i++){
// 找到需要后移的索引
if(x<a[i])break;
}
// 通过数组复制实现数组后移
if (a.length - 1 - i >= 0) System.arraycopy(a, i, a, i + 1, a.length - 1 - i);
a[i]=x;
}