hdu 5096 ACM Rank(treap)

模拟一个acm比赛的排名， 并且支持询问排在第k个的是哪个队和第i个队排第几

丧心病狂， 第一次见1W个队的ACM。

看起来是平衡树， 不过好多细节。。。

首先对于每个提交， 先判断是不是有效的，如果是有效的话就看是不是ac， 如果不是ac的话对应队伍的对应题目要加罚时， 如果ac了的话对应队伍的总时间要加上用时t和对应题目的罚时， 然后做的题目要加1， 这里为了方便就不是题目加1， 而是把总用时减去一个很大的数(1e7)，然后在treap里面排的时候就是按照总时间来排， 然后ac了的话要更新最后提交时间和最后ac时间， 没有ac的话就要更新最后提交时间， 比较蛋疼的就是罚时相等的时候， 要按照它说的规则， 于是就在treap的节点里面加个set， 然后搞一下仿函数。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <queue>
#include <map>
#include <string>
#include <set>
#include <cstdlib>
#include <algorithm>
using namespace std;

#define mnx 20002
#define N 3020
#define E 1000200
#define inf 0x3f3f3f3f
#define ls (p[i].ch[0])
#define rs (p[i].ch[1])
#define md ((ll + rr) >> 1)
const int M = 10000000;



bool acc[mnx][11];
int pen[mnx][11]; // 未提交时候的罚时
int lst[mnx];
int cnt[mnx]; // 总罚时
int lstacc[mnx];

int n, k;

struct ff {
    bool operator()(const int &a, const int &b) const{
        if(lstacc[a] == -1 && lstacc[b] == -1) return a < b;
        if(lstacc[a] == -1 || lstacc[b] == -1) return lstacc[b] == -1;
        return lstacc[a] < lstacc[b];
    }
};
struct node {
    int ch[2], r, s, v, cnt;
    set<int, ff> vv;
    node() {}
    void setIt(int v, int id) {
        this -> v = v;
        vv.clear();
        vv.insert(id);
        ch[0] = ch[1] = 0, r = rand(), s = 1;
        cnt = 1;
    }
}p[mnx * 22];
int sz;
int creat(int v, int id) {
    ++ sz;
    p[sz].setIt(v, id);
    return sz;
}
int cmp(int i, int v) {
    if(p[i].v == v) return -1;
    return v < p[i].v? 0: 1;
}
void mt(int i) {
    p[i].s = p[i].cnt;
    if(ls) p[i].s += p[ls].s;
    if(rs) p[i].s += p[rs].s;
}
void rt(int &i, int d) {
    int k = p[i].ch[d^1];
    p[i].ch[d^1] = p[k].ch[d];
    p[k].ch[d] = i;
    mt(i), mt(k);
    i = k;
}
void ins(int &i, int id, int v) {
    if(!i) {
        i = creat(v, id);
        return;
    }
    int d = cmp(i, v);
    if(d == -1) {
        p[i].cnt ++, p[i].s++;
        p[i].vv.insert(id);
        return;
    }
    ins(p[i].ch[d], id, v);
    if(p[p[i].ch[d]].r > p[i].r)
        rt(i, d ^ 1);
    mt(i);
}
set<int, ff>::iterator it;
void Remove(int &i, int id, int v) {
    int d = cmp(i, v);
    if(d == -1) {
        if(p[i].cnt > 1) {
            p[i].cnt--, p[i].s--;
            p[i].vv.erase(id);
            return;
        }
        if(!ls) i = rs;
        else if(!rs) i = ls;
        else {
            int d2 = p[ls].r > p[rs].r? 1: 0;
            rt(i, d2);
            Remove(p[i].ch[d2], id, v);
        }
    }
    else Remove(p[i].ch[d], id, v);
    if(i) mt(i);
}
int kth(int i, int k) {
    if(!i) return -1;
    int d = k - p[ls].s;
    if(d <= 0) return kth(ls, k);
    if(d <= p[i].cnt) {
        if(d != 1) return -1;
        return *(p[i].vv.begin());
    }
    return kth(rs, d - p[i].cnt);
}
int ran(int i, int id, int v) {
    int d = cmp(i, v);
    int c = 0;
    if(ls) c += p[ls].s;
    if(d == -1) return c + 1;
    if(d == 0) return ran(ls, id, v);
    return c + p[i].cnt + ran(rs, id, v);
}
int readint() {
    char c;
    while((c = getchar()) && !(c >= '0' && c <= '9'));
    int ret = c - '0';
    while((c = getchar()) && c >= '0' && c <= '9')
        ret = ret * 10 + c - '0';
    return ret;
}
char readc() {
    char c;
    while((c = getchar()) && !isalpha(c));
    return c;
}
int main() {
    while(scanf("%d%d", &n, &k) != EOF) {
        int kk = 0;
        memset(pen, 0, sizeof pen);
        memset(lst, -1, sizeof lst);
        memset(cnt, 0, sizeof cnt);
        memset(acc, 0, sizeof acc);
        memset(lstacc, -1, sizeof lstacc);
        int root = 0;
        char op[20];
        sz = 0;
        for(int i = 0; i < n; ++i) {
            ins(root, i, 0);
        }
        while(scanf("%s", op) && op[0] != 'C') {
            ++kk;
            if(op[0] == 'S') {
                int t, team, res;
                int id;
                t = readint();
                team = readint();
                id = readc() - 'A';
                res = readint();
                if(lst[team] != -1 && t - lst[team] < 5) continue;
                if(acc[team][id]) continue;
                if(res == 0) {
                    pen[team][id] += 20;
                    lst[team] = t;
                    continue;
                }
                printf("[%d][%c]\n", team, (char)id + 'A');
                Remove(root, team, cnt[team]);
                cnt[team] += pen[team][id];
                cnt[team] += t;
                acc[team][id] = 1;
                cnt[team] -= M;
                lst[team] = t;
                lstacc[team] = kk;
                ins(root, team, cnt[team]);
                continue;
            }
            int team;
            if(op[0] == 'R') {
                scanf("%d", &team);
                printf("%d\n", ran(root, team, cnt[team]));
                continue;
            }
            int k;
            scanf("%d", &k);
            printf("%d\n", kth(root, k));
        }
        scanf("%s", op);
        puts("");
    }
    return 0;
}