zoj 3612 （multiset）

最笨的方法是自己实现红黑树然后标记count，这样在O(logn)时间实现kth。当然我们不需要造轮子，因为已经有multiset这种东西，稍微拿来改一改就可以了，由于只需要考虑第size/2个数，效率可以分摊到O(1)。当然我们还需要知道，multiset有两条重要的性质：1、相同的数往右子树插入，2、排序是稳定的——也就是插入和删除时不会更改相同键值Item的前序遍历顺序。这样就够了。

#include <cstdio>
#include <cstring>
#include <iostream>
#include <queue>
#include <set>
#include <algorithm>
#include <vector>
using namespace std;
#define LL long long
#define CL(a, val) memset(a, val, sizeof(a))
typedef pair<int, int> PII;
#define MOD 1000003

multiset<int> s;

multiset<int>::iterator it; 
multiset<int>::iterator it1; 
multiset<int>::iterator it2; 

void print_item()
{
    if(s.size() == 0)
        printf("Empty!\n");
    else if(s.size() & 1)
        printf("%d\n", *it1);
    else
    {
        it2 = it1;
        ++ it2;
        LL ans = (LL)*it1 + (LL)*it2;
        if(ans & 1)
        {
            printf("%.1f\n", ans / 2.00);
        }
        else
        {
            printf("%lld\n", ans / 2);
        }
    }
}

void add_item(int x)
{
    s.insert(x);
    if(s.size() == 1)
    {
        it1 = s.begin();
    }
    else if(s.size() & 1)
    {
        if(x >= *it1)
            ++ it1;
    }
    else
    {
        if(x < *it1)
            -- it1;
    }
}

int rem_item(int x)
{
    it = s.find(x);
    if(it == s.end())
        return 1;
    if(x == *it1)
    {
        it = it1;
        if(s.size() & 1)
            -- it1;
        else 
            ++ it1;
    }
    else
    {
        if(x > *it1 && (s.size() & 1))
            -- it1;
        if(x < *it1 && !(s.size() & 1))
            ++ it1;
    }
    s.erase(it);
    return 0;
}

void cd_test()
{
    int n;
    scanf("%d", &n);
    s.clear();
    for(int i = 0; i < n; ++ i)
    {
        char sz[10];
        int op;
        scanf("%s%d", sz, &op);
        if(sz[0] == 'r')
        {
            int rt = rem_item(op);
            if(rt)
            {
                printf("Wrong!\n");
                continue;
            }
        }
        else
        {
            add_item(op);
        }
        print_item();
    }
}

int main() 
{
#if _MSC_VER == 1700
    freopen("in.txt", "r", stdin);
    freopen("out.txt", "w", stdout);
#endif
    int _t;
    scanf("%d", &_t);
    for(int i = 0; i < _t; ++ i)
    {
        cd_test();
    }

    return 0;
}