快速选择第k小的代码练习C++

#include<iostream>
using namespace std;
const int N=100010;

int q[N];

int quick_sort(int q[],int l,int r,int k) {
	if(l>=r)
		return q[l];
	int i=l-1,j=r+1,x=q[l+r>>1];
	while(i<j) {
		do i++;
		while(q[i]<x);
		do j--;
		while(q[j]>x);
		if(i<j)
			swap(q[i],q[j]);
	}

	if(j-l+1>=k)
		return quick_sort(q,l,j,k);
	else
		return quick_sort(q,j+1,r,k-(j-l+1));
}

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

这段代码实现了一个快速排序算法，并找到排序后第k小的元素。

首先，定义了一个常量N为100010，表示数组q的最大长度。

然后，定义了一个全局数组q。

接下来，定义了一个函数quick_sort，该函数用于实现快速排序算法。参数l表示当前排序区间的左边界，r表示当前排序区间的右边界，k表示要找的第k小的元素。函数返回排好序的数组q中第k小的元素。

在函数内部，先判断如果左边界大于等于右边界，则直接返回q[l]，表示当前只有一个元素，无需再排序。

接着，定义了两个指针i和j，分别指向排序区间的左边界和右边界，并找到数组中间位置的元素x（使用位运算右移1位的方式）。这里使用了双指针的方式进行排序。

然后，进入一个循环，循环条件是i<j。在循环中，先将i右移，直到找到一个大于x的元素为止；然后将j左移，直到找到一个小于x的元素为止；接着判断如果i<j，就交换q[i]和q[j]的值。

循环结束后，根据j-l+1（即左边界到j的距离加1）与k的大小关系，判断应该继续在左边界到j的区间内找第k小的元素，还是在j+1到右边界的区间内找第k-(j-l+1)小的元素。然后递归调用quick_sort函数。

最后，主函数中从输入中读取n和k的值，然后循环读取n个元素到数组q中。调用quick_sort函数，打印结果。

整体思路是通过快速排序算法将数组排好序，然后根据k的值返回第k小的元素。