郭大侠与线上游戏

最新推荐文章于 2017-05-02 16:30:00 发布

原创最新推荐文章于 2017-05-02 16:30:00 发布 · 365 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#set

set 专栏收录该内容

1 篇文章

订阅专栏

本文介绍了一种利用两个Set容器来维护数据流中位数的方法，该方法适用于在线更新数据集并快速获取当前中位数值的场景。通过案例演示了如何在插入、删除操作后调整数据分布以保持中位数的准确性。

http://acm.uestc.edu.cn/#/problem/show/1339
本来想用线段树+数据离散化做，结果过不了，所以改用两个set容器，在要求的数分开保存。

#include<algorithm>
#include <iostream>
#include  <cstring>
#include   <cstdio>
#include    <queue>
#include      <set>
#define Max 100005
using namespace std;
queue<int>A;
set<int>big;
int op, key;
set<int> s;
void compare() {
    if (s.size() < big.size() - 1 ) {
        s.insert(*big.begin());
        big.erase(*big.begin());
    }
    else if (s.size() > big.size()) {
        big.insert(*s.rbegin());
        s.erase(*s.rbegin());
    }
}
int main() {
    int N;
    scanf("%d", &N);
    while (N--) {
        scanf("%d", &op);
        switch (op) {
        case 1:
            scanf("%d", &key);
            A.push(key);
            if (big.empty() || key > *big.begin()) {
                big.insert(key);
            }else{
                s.insert(key);
            }
            compare();
            break;
        case 2:
            key = A.front();
            A.pop();
            if (key > *s.rbegin()) {
                big.erase(big.find(key));
            }
            else{
                s.erase(s.find(key));
            }
            compare();
            break;
        case 3:
            printf("%d\n", *big.begin());
        }
    }
    return 0;
}