算法基础---快速选择算法

快速选择算法

问题

使用快速排序算法求出数列从小到大排序后的第k个数。
输入格式：
第一行两个整数： n（数列中数据个数）和 k（第k个数）
第二行输入数列： 包含n个整数

输入样例：

5 2
2 3 4 1 7

输出样例：

2

解题思路

• 找到排序算法递归出口
• 快速排序分治思想，将区间划分。
• 比较 k 与 区间的大小，若 k 小于等于左半区间则递归左半区间，反之则递归右半区间。

#include<iostream>
using namespace std;

int n, k;
const int N = 1e5;
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);
    else return quick_sort(j + 1, r, k - sl);
}

int main(){
    cin >> n >> k;
    for(int i = 0; i < n; i ++)scanf("%d",&q[i]);
    cout << quick_sort(0, n-1, k) << endl;
    return 0;
}