FJUT 奇怪的数列（线性选择算法）题解

题意：找出无需数列中位数（偶数为两个中位数平均数向下取整）

思路：用nth_element（a + first，a + k，a+ end + 1）找出中位数，复杂度一般为O(n)。这个STL能将 [ a + first，a+ end + 1）数组中第k小的数字放在a + k这个位置上，并且k前都比他小，后面都比他大。向下取整应该用 >>1，不要用”/2”  “/2”是向零取整。

代码：

#include<set>
#include<map>
#include<stack>
#include<cmath>
#include<queue>
#include<vector>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
typedef long long ll;
const int maxn = 1e6 + ;
const int seed = ;
const ll MOD = 1e9 + ;
const int INF = 0x3f3f3f3f;
using namespace std;
ll a[maxn];
int main(){
    int n;
    scanf("%d", &n);
    for(int i = ; i <= n; i++)
        scanf("%lld", &a[i]);
    ll ans;
    if(n & ){
        nth_element(a + , a + n /  + , a + n + );
        ans = a[n /  + ];
    }
    else{
        nth_element(a + , a + n / , a + n + );
        ans = a[n / ];
        nth_element(a + , a + n /  + , a + n + );
        ans += a[n /  + ];
        ans >>= ;
    }
    printf("%lld\n", ans);
    return ;
}