快速排序，第k小数目（洛谷P1923）

快排代码

#include <iostream>
#include <cstdio>
using namespace std;
int a[100005],n;

void quick_sort(int l,int r)
{
	int i=l,j=r,flag = a[l+r>>1];
	do{
		while(a[i]<flag) i++;
		while(a[j]>flag) j--;
		if(i<=j){
			swap(a[i],a[j]);
			i++;j--;
		} 
	}while(i<=j);
	if(l<j) quick_sort(l,j);
	if(i<r) quick_sort(i,r); 
} 

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

第k小数

#include <iostream>
#include <cstdio>
using namespace std;
int a[5000010],n,ans,k;


void findkth(int l,int r)
{
	if(l==r){
		ans = a[l];
		return ;
	}
	int i=l,j=r,flag = a[l+r>>1];
	do{
		while(a[i]<flag) i++;
		while(a[j]>flag) j--;
		if(i<=j){
			swap(a[i],a[j]);
			i++;j--;
		}
	}while(i<=j);
	
	if(k<=j) findkth(l,j);
	else if(k>=i) findkth(i,r);
	else findkth(j+1,i-1);
}

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