面試 | 程式猿面試,Elasticsearch被坑被虐的體無完膚...
阿新 • • 發佈:2020-11-24
給定一個長度為n的整數數列,以及一個整數k,請用快速選擇演算法求出數列從小到大排序後的第k個數。
輸入格式
第一行包含兩個整數 n 和 k。
第二行包含 n 個整數(所有整數均在1~109109範圍內),表示整數數列。
輸出格式
輸出一個整數,表示數列的第k小數。
資料範圍
1≤n≤1000001≤n≤100000,
1≤k≤n1≤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, intr, 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; }