【深基9.例4】求第 k 小的数

一、题目描述

输入 $n$（$1\le n<5000000$ 且 $n$ 为奇数）个数字 $a_i$（$1\le a_i<{10}^9$），输出这些数字的第 $k$ 小的数。最小的数是第 $0$ 小。

请尽量不要使用 nth_element 来写本题，因为本题的重点在于练习分治算法。

二、样例输入 #1

5 1
4 3 2 1 5

三、样例输出 #1

2

四、错误经历

使用了快速排序的标准模板，超时了

import java.util.*;

public class Main{
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int n = sc.nextInt();
        int k = sc.nextInt();
        int[] a = new int[n];
        for (int i = 0; i < n; i++) {
            a[i] = sc.nextInt();
        }
        quickSort(a, 0, n - 1);
        System.out.println(a[k]);
    }

    public static void quickSort(int[] a, int low, int high) {
        int i,j;
        int temp,t;
        if (low > high){
            return;
        }
        i = low;
        j = high;
        //temp就是基准位
        temp = a[low];

        while (i < j){
            //先看右边，依次往左递减
            while (a[j] >= temp && j > i){
                j--;
            }
            //再看左边，依次往右递增
            while (a[i] <= temp && i < j){
                i++;
            }
            //如果满足条件则交换
            if (i < j){
                t = a[i];
                a[i] = a[j];
                a[j] = t;
            }
        }
        //最后将基准为与i和j相等位置的数字交换
        a[low] = a[i];
        a[i] = temp;
        //递归调用左半数组
        quickSort(a,low,j - 1);
        //递归调用右半数组
        quickSort(a,j + 1,high);
    }
}

import java.util.*;

public class Main{
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int n = sc.nextInt();
        int k = sc.nextInt();
        int[] a = new int[n];
        for (int i = 0; i < n; i++) {
            a[i] = sc.nextInt();
        }
        Quick_Sort(a, 0, n - 1,k);
        System.out.println(a[k]);
    }

    public static void Quick_Sort(int arr[], int begin, int end,int k){
        if(begin > end)
            return;
        int tmp = arr[begin];
        int i = begin;
        int j = end;
        while(i != j){
            while(arr[j] >= tmp && j > i)
                j--;
            while(arr[i] <= tmp && j > i)
                i++;
            if(j > i){
                int t = arr[i];
                arr[i] = arr[j];
                arr[j] = t;
            }
        }
        arr[begin] = arr[i];
        arr[i] = tmp;
        if(k == i || k == j) {
            return;
        } else if(k < i){
            Quick_Sort(arr, begin, i-1,k);
        }else if(k > j){
            Quick_Sort(arr, j+1, end,k);
        }else {
            Quick_Sort(arr, i, j,k);
        }
    }
}

直接使用了Arrays.sort( )，也超时了

import java.util.*;

public class Main{
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int n = sc.nextInt();
        int m = sc.nextInt();
        int[] a = new int[n];
        for (int i = 0; i < n; i++) {
            a[i] = sc.nextInt();
        }
        Arrays.sort(a);
        System.out.println(a[m]);
    }
}

我废了，栓Q