洛谷3871 [TJOI2010]中位数 维护队列的中位数

题目描述

给定一个由N个元素组成的整数序列,现在有两种操作:

1 add a

在该序列的最后添加一个整数a,组成长度为N + 1的整数序列

2 mid 输出当前序列的中位数

中位数是指将一个序列按照从小到大排序后处在中间位置的数。(若序列长度为偶数,则指处在中间位置的两个数中较小的那个)

例1:1 2 13 14 15 16 中位数为13

例2:1 3 5 7 10 11 17 中位数为7

例3:1 1 1 2 3 中位数为1

输入输出格式

输入格式: 

第一行为初始序列长度N。第二行为N个整数,表示整数序列,数字之间用空格分隔。第三行为操作数M,即要进行M次操作。下面为M行,每行输入格式如题意所述。

输出格式:

对于每个mid操作输出中位数的值

输入输出样例

输入样例#1: 
6
1 2 13 14 15 16
5
add 5
add 3
mid
add 20
mid
输出样例#1: 
5
13

说明

对于30%的数据,1 ≤ N ≤ 10,000,0 ≤ M ≤ 1,000

对于100%的数据,1 ≤ N ≤ 100,000,0 ≤ M ≤ 10,000

序列中整数的绝对值不超过1,000,000,000,序列中的数可能有重复

每个测试点时限1秒

解法

维护中位数,两个堆

发现题解大神有拿 sort+二分查找 暴力出奇迹A题的 , 还有主席树的。

操作就是 

你需要一个大根堆和一个小根堆,小根堆里面放前 N/2 个大的数,大根堆里面放后 N-N/2 大的数 

这样就能保证堆顶就是 中位数

在插入的时候,把 大于 堆顶的扔到小根堆  否则扔到大根堆

当然 我们还要维护 使得两个堆大小最多差1 

废话

懒得写堆当然要用priority了 

小根堆写法priority_queue<int,vector<int>,greater<int> >  或者你也可以机智的把数先变成负的直接扔进priority_queue里当小根堆用

#include<bits/stdc++.h>
using namespace std;
int N,Q,cnt;
char s[10];
priority_queue <int> q1,q2;
int add()
{
    cnt++;
    int x;
    scanf("%d",&x);
    if(q2.empty())q2.push(x);
    else if(x<q2.top())q2.push(x);
    else q1.push(-x);
    while(q2.size()>cnt-cnt/2){
        int tmp=-q2.top();q2.pop();
        q1.push(tmp); 
    }
    while(q1.size()>cnt/2){
        int tmp=-q1.top();q1.pop();
        q2.push(tmp);
    }
} 
int main()
{
    scanf("%d",&N);
    for(int i=1;i<=N;i++)add();
    scanf("%d",&Q);
    while(Q--){
        scanf("%s",s);
        if(s[0]=='a')add();
        else printf("%d
",q2.top()); 
    }     
    return 0;
} 

吐槽

我永远也不想回到教室学习电磁感应和基因突变

原文地址:https://www.cnblogs.com/Elfish/p/7905853.html