POJ 2388 Who's in the Middle (快速选择算法:O(N)求数列第K大)

时间:2023-03-08 17:46:27

【题意】求数列中间项。

---这里可以扩展到数列第K项。

第一次做的时候直接排序水过了= =……这一次回头来学O(N)的快速选择算法。

快速选择算法基于快速排序的过程,每个阶段我们选择一个数为基准,并把区间划分成小于这个数和大于这个数的两个子区间,此时便可以判断这个数是不是第k大项,如果比K大,则去左区间找,否则去右区间找。

#include
#include
#include
#include
#include
using namespace std;

template
doubleORint QuickSelect(doubleORint A[], int head, int tail, int k){
int i = head, j = tail;
doubleORint x = A[head];
while(i x) j --;
swap(A[i], A[j]);
while(A[i] k){
return QuickSelect(A, head, i-1, k);
}
else{
return QuickSelect(A, i+1, tail, k - (i - head + 1));
}
}
int a[10005];
int main(){
int n;
scanf("%d", &n);
for (int i = 0; i