zoukankan      html  css  js  c++  java
  • 【POJ3784】Running Median

    动态维护中位数的题目。

    我们采用对顶堆做法,建立一个大根堆存储前半段序列(排序后),小根堆存储后半段序列,通过维护两个堆使得他们元素个数之差不大于1 ,这样这个序列的中位数就是大根堆的堆顶。

    关于如何维护对顶堆:如果两个堆元素个数差大于1,我们就把大根堆的堆顶放到小根堆当中(或是把小根堆堆顶放到大根堆当中)直到两个堆“平衡”为止。

    ps:当堆空时调用top会导致RE,所以调用top前务必保证堆中有元素。

     1 #include <iostream>
     2 #include <cstdio>
     3 #include <cstring>
     4 #include <algorithm>
     5 #include <queue>
     6 #include <vector>
     7 using namespace std;
     8 vector<int> ans;
     9 priority_queue<int>minn;
    10 priority_queue<int,vector<int>,greater<int> >maxx;
    11 int T,n,id;
    12 inline int read() {
    13     int ret=0;
    14     int op=1;
    15     char c=getchar();
    16     while(c<'0'||c>'9') {if(c=='-') op=-1; c=getchar();}
    17     while(c<='9'&&c>='0') ret=ret*10+c-'0',c=getchar();
    18     return ret*op;
    19 }
    20 int main() {
    21     cin>>T;
    22     while(T--) {
    23         cin>>id>>n;
    24         while(maxx.size()) maxx.pop();
    25         while(minn.size()) minn.pop();
    26         ans.clear();
    27         for(int i=1,x;i<=n;i++) {
    28             cin>>x;
    29             if(maxx.empty()) maxx.push(x);
    30             else if(x>maxx.top()) maxx.push(x);
    31             else minn.push(x);
    32             while(maxx.size()<minn.size()) {
    33                 maxx.push(minn.top());
    34                 minn.pop();
    35             } 
    36             while(maxx.size()-minn.size()>1) {
    37                 minn.push(maxx.top());
    38                 maxx.pop();
    39             }
    40             if(i%2) ans.push_back(maxx.top());
    41         }
    42         printf("%d %d
    ",id,n/2+1);
    43 //        puts("OK");
    44         for(int i=0;i<ans.size();i++) {
    45             if(i&&i%10==0) puts("");
    46             if(i%10) putchar(' ');
    47             printf("%d",ans[i]);
    48         }
    49         puts("");
    50     }
    51     return 0;
    52 }
    AC Code
  • 相关阅读:
    程序员学习能力提升三要素(转)
    网页游戏框架
    CS心得
    项目主管谈网页游戏:一将功成万骨枯
    分享成为高效程序员的7个重要习惯
    看代码的实质
    点击delphi中wwDBGrid标题进行排序
    DataRow 学习
    学习遇到的问题MVC设置 起始页后怎样恢复
    泛型中的where
  • 原文地址:https://www.cnblogs.com/shl-blog/p/10962152.html
Copyright © 2011-2022 走看看