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

【题意】求数列中间项。 ---这里可以扩展到数列第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 < j){
        while(A[j] > x) j --;
        swap(A[i], A[j]);
        while(A[i] < x) i ++;
        swap(A[i], A[j]);
    }
    if (i - head == k){
        return A[i];
    }
    else if (i - head > 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 < n; i ++){
        scanf("%d", &a[i]);
    }
    printf("%d", QuickSelect(a, 0, n-1, n/2));
    return 0;
}