UESTC 1339 郭大侠与线上游戏 (STL插入删除/迭代器)

STL erase后指针变化

• 对于用数组实现的数据结构(vector,deque),除一个元素后之后所有的迭代器都会失效
• 对于关联型数据结构,删除后只会对当前节点有影响
• 删除节点前应该先取得下一个节点或上一个节点的位置

for( itList = List.begin(); itList != List.end(); ) {
    if(shouldDel( *itList))
        itList = List.erase( itList);
        //List.erase( itList++);  set/map/..
    else itList++;
}

题意

每次在队列后加入一个值或删除队列首部一个值，或者询问队列中的数的中位数

思路

一开始还在想，要是有自动排序又能瞬间取出中值的数据结构就好了，结果发现操作很简单，可以用stl来完成

代码

需要考虑删除的是当前迭代器指向的元素的情况

#include <bits/stdc++.h>

#define mem(a,b) memset(a,b,sizeof(a))
#define rep(i,a,b) for(int i=a;i<b;i++)
#define sc(a) scanf("%d",&a)
const int INF=0x3f3f3f3f;
const int maxn=1e6+50;
const int mod=1e9+7;
const double eps=1e-8;
#define pii pair<int,int>
typedef long long ll;
typedef unsigned int ui;
using namespace std;

int main()
{
#ifndef ONLINE_JUDGE
    //freopen("in.txt","r",stdin);
    //freopen("out.txt","w",stdout);
#endif
    int n;
    while(~sc(n)){
        set<int> se;
        queue<int> qu;
        set<int>::iterator itm,ite;
        rep(i,0,n){
            int op,x; sc(op);
            if(op==1){
                sc(x);
                qu.push(x);
                se.insert(x);
                int sz=se.size();
                if(sz==1) itm=se.begin();
                else{
                    int cur=*itm;
                    if(x<cur && (sz&1)) itm--;
                    else if(x>cur && !(sz&1)) itm++;
                }
            }else if(op==2){
                x=qu.front();
                qu.pop();
                ite=se.find(x);
                int cur=*itm;
                int sz=se.size();
                if(x<cur && (sz&1)) itm++;
                else if(x>cur && !(sz&1)) itm--;
                else if(x==cur && (sz&1)) itm++;
                else if(x==cur && !(sz&1)) itm--;
                se.erase(ite);
            }else{
                printf("%d\n",*itm);
            }
        }
    }
    return 0;
}