[洛谷P1168] 中位數

題目描述

給出一個長度爲N的非負整數序列A[i]，對於全部1 ≤ k ≤ (N + 1) / 2，輸出A[1], A[3], …, A[2k - 1]的中位數。即前1，3，5，……個數的中位數。

輸入輸出格式

輸入格式：

輸入文件median.in的第1行爲一個正整數N，表示了序列長度。

第2行包含N個非負整數A[i] (A[i] ≤ 10^9)。

輸出格式：

輸出文件median.out包含(N + 1) / 2行，第i行爲A[1], A[3], …, A[2i – 1]的中位數。

輸入輸出樣例

輸入樣例#1： 複製

7
1 3 5 7 9 11 6

輸出樣例#1： 複製

1
3
5
6

說明

對於20%的數據，N ≤ 100；

對於40%的數據，N ≤ 3000；

對於100%的數據，N ≤ 100000。

思路

權值線段樹；

代碼實現

 1 #include<cstdio>
 2 #include<algorithm>
 3 const int maxn=1e5+10;
 4 int n,m;
 5 int s[maxn],d[maxn];
 6 int t[maxn<<2];
 7 void add(int k,int l,int r,int x){
 8     if(l==r){t[k]++;return;}
 9     int mid=l+r>>1,ls=k<<1,rs=ls|1;
10     if(x<=d[mid]) add(ls,l,mid,x);
11     else add(rs,mid+1,r,x);
12     t[k]=t[ls]+t[rs];
13 }
14 int find(int k,int l,int r,int x){
15     if(l==r) return d[l];
16     int mid=l+r>>1,ls=k<<1,rs=ls|1;
17     if(x<=t[ls]) return find(ls,l,mid,x);
18     else return find(rs,mid+1,r,x-t[ls]);
19 }
20 int main(){
21     scanf("%d",&n);
22     for(int i=1;i<=n;i++) scanf("%d",&s[i]),d[i]=s[i];
23     std::sort(d+1,d+n+1);
24     for(int i=1;i<=n;i++) if(d[i]!=d[i+1]) d[++m]=d[i];
25     for(int i=1;i<=n;i++){
26         add(1,1,m,s[i]);
27         if(i&1) printf("%d\n",find(1,1,m,i+1>>1));
28     }
29     return 0;
30 }