HDU_5249(百度之星D题)

来源：互联网发布：300451创业软件股吧编辑：程序博客网时间：2024/05/18 00:01

因为之前没写过平衡树的题，所以很自然地只会用set来写。。。然后，很蠢地想直接找set容器中间位置的那个值，结果iterator没有重载+唉。。。翻了一下AC的代码（果然有跟我一样用set来写的），然后发现是两个set容器解决了这个问题。。。其实很容易想到，一个set容器放前一半的数，一个set容器放后一半的数，就行了。。。

#include <iostream>#include <cstdio>#include <cstring>#include <cmath>#include <algorithm>#include <cstdlib>#include <set>#include <queue>#define FOR(i,x,y)  for(int i = x;i < y;i ++)#define IFOR(i,x,y) for(int i = x;i > y;i --)using namespace std;int n;int main(){    //freopen("test.in","r",stdin);    int tCase = 0;    while(~scanf("%d",&n)){        printf("Case #%d:\n",++tCase);        int num;        queue <int> s;        set <int>   p,q;        char str[10];        while(n--){            getchar();            scanf("%s",str);            if(str[0] == 'q'){                printf("%d\n",*(q.begin()));            }            else if(str[0] == 'i'){                scanf("%d",&num);                s.push(num);                if(q.empty() || num < *(q.begin())){                    p.insert(num);                }                else                    q.insert(num);            }            else{                int v = s.front(); s.pop();                if(p.find(v) != p.end()){                    p.erase(v);                }                else                    q.erase(v);            }            while(p.size() > q.size()){                set <int> :: iterator it = p.end();                it--;                int v = *it;                p.erase(it);                q.insert(v);            }            while(q.size() > p.size()+1){                set <int> :: iterator it = q.begin();                int v = *it;                p.insert(v);                q.erase(it);            }        }    }    return 0;}

0 0