找到無序陣列中最小的k個數

敲響指間的藝術發表於2020-12-21
陣列
輸入包含兩行，第一行包含兩個整數n和k(1≤k≤n≤105)，代表陣列arr的長度，第二行包含n個整數，代表陣列arr。

輸出描述:

輸出包含一行，k個整數，代表陣列中最小的k個整數。

備註:

時間複雜度O(nlog_2k)和O(n)O(nlog2​k)和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;
}

 

相關文章