1. 程式人生 > >FJUT 奇怪的數列(線性選擇算法)題解

FJUT 奇怪的數列(線性選擇算法)題解

mat 並且 ace 位置 string lse 選擇 nth max

題意:找出無需數列中位數(偶數為兩個中位數平均數向下取整)

思路:用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 + 10; const int seed = 131; const ll MOD = 1e9 + 7; const int INF = 0x3f3f3f3f; using namespace std; ll a[maxn]; int main(){ int n; scanf(
"%d", &n); for(int i = 1; i <= n; i++) scanf("%lld", &a[i]); ll ans; if(n & 1){ nth_element(a + 1, a + n / 2 + 1, a + n + 1); ans = a[n / 2 + 1]; } else{ nth_element(a + 1, a + n / 2, a + n + 1); ans = a[n / 2]; nth_element(a
+ 1, a + n / 2 + 1, a + n + 1); ans += a[n / 2 + 1]; ans >>= 1; } printf("%lld\n", ans); return 0; }

FJUT 奇怪的數列(線性選擇算法)題解