中位数(洛谷p1168)(堆/树状数组+二分/线段树+二分)

题目地址

注意偶数的情况

堆

构建两个堆，一个大根堆，一个小根堆，每次加入的时候，维护这两个堆，使得大根堆的个数在一定情况下比小根堆多1或者相等。 大根堆的最后一个元素即为中位数
当上一次处理完：
大根堆与小根堆个数相等。则下一次插入一定插入到大根堆。
此时判断，小根堆的 堆顶>x 则将小根堆的堆顶元素插入到大根堆，小根堆pop，再把x压入小根堆。否则直接压入大根堆。
大根堆与小根堆个数不相等与上面类似

代码

#include<iostream>
#include<algorithm>
#include<queue>
using namespace std;
const int N = 1e5 + 10;
typedef long long ll;

int a[N];
priority_queue<int,vector<int>,greater<int>> small;
priority_queue<int,vector<int>,less<int>> big;

void work(int x){
	int lens = (int)small.size(),lenb = (int)big.size();
	if(lens == lenb){
		if(x <= small.top()){
			big.push(x);
		}else{
			big.push(small.top());
			small.pop();
			small.push(x);
		}
	}else{
		if(x >= big.top()){
			small.push(x);
		}else{
			small.push(big.top());
			big.pop();
			big.push(x);
		}
	}
}

int main(){
	#ifdef ddgo
		freopen("C:\\Users\\asus\\Desktop\\ddgoin.txt","r",stdin);
    #endif
    ios::sync_with_stdio(0);cin.tie(0);cout.tie(0);
    int n; cin>>n;
    for(int i=1;i<=n;i++) cin>>a[i];
    cout<<a[1]<<endl;
    big.push(a[1]);
    if((n & 1) == 0) n --;
    for(int i=2;i<=n;i+=2){
    	work(a[i]); work(a[i+1]);
    	cout<<big.top()<<endl;
	}
    return 0;
}

树状数组+二分

每插入一个数，将其个数加1。二分答案，判断比当前元素小的出现的个数是否小于中位数。
树状数组与线段树都是维护区间值。二分里面询问1~mid的区间石否出现的次数，找到小于中位数的最大的数。 小于是为了，当中位数这个个数出现很多次，不会有等于的情况发生。

代码：

#include<iostream>
#include<algorithm>
#include<vector>
using namespace std;
const int N = 1e5 + 10;
typedef long long ll;

int c[N],n,a[N];
vector<int> b;

int find(int x){
	return lower_bound(b.begin(),b.end(),x)-b.begin();
}

#define lowbit(x) (x & -x)
void update(int x,int v){
	for(int i=x;i<=n;i+=lowbit(i)) c[i] += v;
}

int ask(int x){
	int res = 0;
	for(int i=x;i;i-=lowbit(i)) res += c[i];
	return res;
}

int main(){
	#ifdef ddgo
		freopen("C:\\Users\\asus\\Desktop\\ddgoin.txt","r",stdin);
    #endif
    ios::sync_with_stdio(0);cin.tie(0);cout.tie(0);
    cin>>n;
    for(int i=1;i<=n;i++) cin>>a[i],b.push_back(a[i]);
    sort(b.begin(),b.end());
    b.erase(unique(b.begin(),b.end()),b.end());
    for(int i=1;i<=n;i++) a[i] = find(a[i]) + 1;
    cout<<b[a[1]-1]<<endl;
    update(a[1],1);
    if((n & 1) == 0) n --;
    for(int i=2;i<=n;i+=2){
    	update(a[i],1);
    	update(a[i+1],1);
    	int l = 1,r = n;
    	while(l < r){
    		int mid = (l + r + 1) >> 1;
    		if(ask(mid) >= (i+2)/2) r = mid - 1;
    		else l = mid;
		}
		cout<<b[l]<<endl;
	}
    
    return 0;
}

线段树+二分

和树状数组维护一样

#include<iostream>
#include<algorithm>
#include<vector>
using namespace std;
const int N = 1e5 + 10;
typedef long long ll;

int n,a[N];
vector<int> b;

#define l(x) t[x].l
#define r(x) t[x].r
#define sum(x) t[x].sum

struct SegmentTree{
	int l,r,sum;
}t[N*4];

int find(int x){
	return lower_bound(b.begin(),b.end(),x)-b.begin();
}

void push_up(int p){
	sum(p) = sum(p*2) + sum(p*2+1);
}

void build(int p,int l,int r){
	l(p) = l,r(p) = r;
	if(l == r) return;
	int mid = (l + r) >> 1;
	build(p*2,l,mid);build(p*2+1,mid+1,r);
}

void update(int p,int x,int v){
	if(l(p) == r(p)){
		sum(p) += v;
		return ;
	}
	int mid = (l(p) + r(p)) >> 1;
	if(x <= mid) update(p*2,x,v);
	else update(p*2+1,x,v);
	push_up(p);
}

int ask(int p,int l,int r){
	if(l <= l(p) && r >= r(p)) return sum(p);
	int val = 0,mid = (l(p) + r(p)) >> 1;
	if(l <= mid) val += ask(p*2,l,r);
	if(r > mid) val += ask(p*2+1,l,r);
	return val;
}

int main(){
	#ifdef ddgo
		freopen("C:\\Users\\asus\\Desktop\\ddgoin.txt","r",stdin);
    #endif
    ios::sync_with_stdio(0);cin.tie(0);cout.tie(0);
    cin>>n;
    for(int i=1;i<=n;i++) cin>>a[i],b.push_back(a[i]);
    sort(b.begin(),b.end());
    b.erase(unique(b.begin(),b.end()),b.end());
    for(int i=1;i<=n;i++) a[i] = find(a[i]) + 1;
    cout<<b[a[1]-1]<<endl;
    build(1,1,n);
    update(1,a[1],1);
    if((n & 1) == 0) n --;
    for(int i=2;i<=n;i+=2){
    	update(1,a[i],1);
    	update(1,a[i+1],1);
    	int l = 1,r = n;
    	while(l < r){
    		int mid = (l + r + 1) >> 1;
    		if(ask(1,1,mid) >= (i+2)/2) r = mid - 1;
    		else l = mid;
		}
		cout<<b[l]<<endl;
	}
    
    return 0;
}