第k個數(排序)

給定一個長度爲n的整數數列,以及一個整數k,請用快速選擇算法求出數列的第k小的數是多少。java

輸入格式

第一行包含兩個整數 n 和 k。算法

第二行包含 n 個整數(全部整數均在1~109109範圍內),表示整數數列。ui

輸出格式

輸出一個整數,表示數列的第k小數。spa

數據範圍

1n1000001≤n≤100000,
1kn1≤k≤n
code

輸入樣例:

5 3
2 4 1 5 3

輸出樣例:



快速排序,時間O(n*logn)3
       import java.util.Scanner;

       public class Main {
           public static void quick_sort(int a[],int l,int r){
                 if(l>=r) return;
                 int i=l-1,j=r+1,x=a[l+r>>1];
                 while(i<j){
                       while(a[++i]<x);
                       while(a[--j]>x);
                       if(i<j){
                           int t=a[i];
                           a[i]=a[j];
                           a[j]=t;
                       }
                 }
                 quick_sort(a,l,j);
                 quick_sort(a,j+1,r);
           }
           public static void main(String[] args) {
                Scanner scan=new Scanner(System.in);
                int n=scan.nextInt();
                int k=scan.nextInt();
                int a[]=new int[n];
                for(int i=0;i<n;i++) a[i]=scan.nextInt();
                quick_sort(a,0,n-1);
                System.out.println(a[k-1]);
        }
}
相關文章
相關標籤/搜索