#include <bits/stdc++.h>

int a[5000005];
int n, k;

int mid(int arr[], int i1, int i2, int i3){
	if(arr[i1] > arr[i2])
		i2 = i1;
	if(arr[i2] > arr[i3])
		i2 = i3;
	return i2;
}

void swap(int &a, int &b){
	int temp = a;
	a = b;
	b = temp;
}

int pivot(int arr[], int lo, int hi){	//[lo, hi)
	//[0, lo) 小于等于pivot
	//[hi, n) 大于等于pivot
	int mi = (lo + hi) >> 1;
	int pr = mid(arr, lo, mi, hi-1);	//pivotRank
	swap(arr[lo], arr[pr]);
	int pivot = arr[lo];
	while(hi > lo){
		while(lo < hi){
			if(hi >= n || arr[hi] > pivot)
				hi--;
			else{
				arr[lo++] = arr[hi];
				break;			
			}
		}
		while(lo < hi){
			if(arr[lo] < pivot)
				lo++;
			else{
				arr[hi--] = arr[lo];
				break;
			}
		}
	}
	arr[lo] =  pivot;
	return lo;
}

void solve(int arr[], int lo, int hi, int k){
	//printf("%d %d\n", lo, hi);
	int x = 0;
	while(k != (x = pivot(arr, lo, hi))){
		if(x < k)
			lo = x+1;
		else
			hi = x;
	}
	printf("%d\n", arr[x]);
}
// void sort(int arr[], int lo, int hi){
// 	// printf("%d %d\n", lo, hi);
// 	// for(int i = 0; i < n; i++)
// 	// 	printf("%d ", a[i]);
// 	// printf("\n");
// 	int k = pivot(arr, lo, hi);
// 	// for(int i = 0; i < n; i++)
// 	// 	printf("%d ", a[i]);
// 	// printf("\n");
// 	if(k - lo > 1)
// 		sort(arr, lo, k);
// 	if(hi - k - 1 > 1)
// 		sort(arr, k+1, hi);
// }

int main(void){
//	FILE * fp = fopen("in.txt", "r");
	scanf("%d %d", &n, &k);
//	fscanf(fp, "%d %d", &n, &k);
	for(int i = 0; i < n; i++){
		scanf("%d", a+i);
//		fscanf(fp, "%d", a+i);
	}
	solve(a, 0, n, k);
	//sort(a, 0, n);
	// for(int i = 0; i < n; i++)
	// 	printf("%d ", a[i]);
}

代码如上，用了同样pivot函数的快排吸个氧气就ac了。但是这段代码却出现了一个tle。