AcWing 786 第k个数 (快速选择算法，快排思想)

题目链接：点击查看

题目描述：

给定一个长度为 n 的整数数列，以及一个整数 m ，请用快速选择算法求出数列从小到大排序后的第 m 个数。

输入输出

输入

5  3

2  4  1  5  3 

输出

3

题目分析 ：

对于本题应采用快速选择算法，快速选择算法是基于快速排序算法思想延伸出的一种算法，可以快速找出区间中第m小的数。其大致方法或思想为：某区间根据元素于基准点大小关系分成两部分，左侧部分的数都比所选取的分界点数字要小，同理右侧部分的数都比所选取的分界点数字要大。因此，我们可以计算左侧区间的长度sl，若sl比m大，则第m小的数一定在左侧区间，再对左侧区间进行左侧排序即可，直到找到此数开始回退。反之，若sl比m小，则第m小的数在右侧区域，此时在整体区间第m小变成右侧区域第m-sl小，在对右侧区间进行递归排序操作。与快排不同的是，在快速选择算法中我们根据情况只对一侧区间进行递归排序操作，而快排左右两个区间都要递归。详见如下代码。

代码：
 

#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
const int N = 1e5 + 7;
int n,m;
int arr[N];
int quick_select(int *arr, int l,int r, int m) {
    if (l >= r)
    return arr[l];
    int i = l - 1, j = r + 1 , k = arr[l];
    while (i < j) {
    	do i++; while (arr[i] < k);
    	do j--; while (arr[j] > k);
    	if (i < j)
    	swap(arr[i], arr[j]);
	}
	int sl = j - l + 1;//分界点左侧区间大小 
	if (m <= sl)  return quick_select(arr, l, j, m);//表示第m小的数在分界点左侧，再对左侧区间进行递归 
	return quick_select(arr, j+1, r,m - sl);//表示第m小的数在分界点右侧，此时在右侧区间中是第m-sl小的数 
} 
int main() {
	scanf("%d %d", &n, &m);
	for (int i = 0; i < n; i ++ ) {
		scanf("%d", &arr[i]);
	}  
	printf("%d",quick_select(arr, 0, n - 1, m)); 
	return 0;
}