洛谷P1168中位数

最新推荐文章于 2024-07-25 21:13:04 发布

火龙果king

最新推荐文章于 2024-07-25 21:13:04 发布

阅读量50

点赞数

分类专栏： NOIP 文章标签：算法 c++

本文链接：https://blog.csdn.net/deffdff/article/details/130296545

版权

NOIP 专栏收录该内容

16 篇文章 0 订阅

订阅专栏

如果用循环+sort的话肯定会爆，所以这里介绍一种简单的方法：

用两个堆，一个大根堆，一个小根堆。

------------------------------- ------------------------------------

1 3 5 7 8 | | 13 16 17 18

------------------------------- ------------------------------------

大根堆小根堆

这样大根堆堆顶就是答案了。

中位数传送门

代码如下：

#include<iostream>
#include<cstdio>
#include<queue>
using namespace std;
const int maxn=100005;
int n;
priority_queue<int> q1;
priority_queue<int,vector<int>,greater<int> > q2;
int main(){
    cin>>n;
    for(int i=1;i<=n;i++){
        int a;
        scanf("%d",&a);
        int s1=q1.size();
        int s2=q2.size();
        if(s1==0) q1.push(a);
        else{
            int a1=q1.top();
            if(a<=a1){
                q1.push(a);
                s1++;
            }
            else{
                q2.push(a);
                s2++;
            }
        }
        while(s1-s2>1){
            q2.push(q1.top());
            q1.pop();
            s1--;
            s2++;
        }
        while(s2-s1>=1){
            s2--;
            s1++;
            q1.push(q2.top());
            q2.pop();
        }
        if(i%2==1) printf("%d\n",q1.top());
    }
    return 0;
}