快排求第k大

/* 设某次排序后，分界点x右边有p个元素， （此时x左边元素皆小于等于x,右边元素皆大于等于x） 若p == k-1，则 x 为恰第 k 大； 若p < k-1，则在区间[left, i) 之间求第 k-p-1 大 若p > k-1，则在区间[i+1, right) 之间求第k大 */ #include <cstdio> int qsort(int* A, int left, int right, int k){//在左闭右开区间[left, right)求第k大 int i = left, j = right - 1; int tmp = A[left]; while(i < j){ while(A[j] >= tmp && i < j) j--;//从右往左第一个比tmp小的数 if(i < j) A[i++] = A[j]; while(A[i] <= tmp && i < j) i++;//从左往右第一个比tmp大的数 if(i < j) A[i] = A[j--]; } A[i] = tmp; int p = right - i - 1;//A[i]右边有p个元素 if(p == k-1) return tmp; if(p < k-1) return qsort(A, left, i - 1, k - p - 1); if(p > k-1) return qsort(A, i + 1, right, k); } int main(){ int n, k; while(scanf("%d%d", &n, &k)==2){ int* A = new int[n+1]; for(int i = 0; i < n; i++) scanf("%d", &A[i]); //int A[10] = {13,16,2,1,-1,6,54,123,32,-90}; int ans = qsort(A, 0, n, k);//求第k大 printf("%d\n", ans); } return 0; }