1. 程式人生 > 實用技巧 >面試 | 程式猿面試,Elasticsearch被坑被虐的體無完膚...

面試 | 程式猿面試,Elasticsearch被坑被虐的體無完膚...

給定一個長度為n的整數數列,以及一個整數k,請用快速選擇演算法求出數列從小到大排序後的第k個數。

輸入格式

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

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

輸出格式

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

資料範圍

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

輸入樣例:

5 3
2 4 1 5 3

輸出樣例:

3
#include <iostream>

using namespace std;

const int N = 100010 ;

int n,k;
int q[N];
int quick_sort (int l, int
r, int k) { if( l >= r ) return q[l]; int x = q[l],i = l - 1, j = r + 1; while (i<j) { while (q[ ++i ]<x); while (q[ --j ]>x); if( i < j ) swap(q[i],q[j]); } int sl = j - l + 1; if( k <= sl ) return quick_sort( l, j, k);
return quick_sort( j + 1, r, k-sl); } int main () { cin >> n >> k; for(int i = 0; i<=n; i++) cin >> q[i]; cout << quick_sort( 0 , n - 1 , k ); return 0; }