洛谷1923-第k小-C-(快速排序+分治算法)

AC代码

#include<stdio.h>
int a[5000000];
int n,k;
int deal(int a[],int l,int r){
	int temp=a[l];
	while(l<r){
		while(l<r && a[r]>=temp){
			r-=1;
		}
		a[l]=a[r];
		while(l<r && a[l]<=temp){
			l+=1;
		}
		a[r]=a[l];
	}
	a[l]=temp;
	return l;
}
void fast_sort(int a[],int l,int r,int k){
	int mid=deal(a,l,r);
	if(mid==k){
		printf("%d",a[k]);
		return ;
	}
	else if(mid>k){
		fast_sort(a,l,mid-1,k);
	}
	else{
		fast_sort(a,mid+1,r,k);
	}
}
int main(){
	scanf("%d %d",&n,&k);
	for(int i=0;i<n;i++){
		scanf("%d",&a[i]);
	}
	fast_sort(a,0,n-1,k);
	return 0;
}