HDU 2852 KiKi's K-Number 主席树

题目：http://acm.hdu.edu.cn/showproblem.php?pid=2852

题意：维护一个数据结构，有三种操作：0 e代表把e加入到数据结构中，1 e代表把e从数据结构中删除，2 a k代表求数据结构中大于a的第k大数

思路：用主席树可做。对于前两种操作，就是主席树中增删节点，都通过新建节点来完成，对于第三种操作，可以求出小于等于a的个数cnt，可以用树状数组来维护小于等于a的个数，然后查询整个数据结构中的第cnt + k大数

总结：写漏了一个计数，哇了两发。。。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
using namespace std;

const int N = 100010;
int n = 100000, tot, cnt;
int root[N], lson[N*20], rson[N*20], val[N*20];
int bit[N], num[N];
void add(int i, int x)
{
    while(i <= n) bit[i] += x, i += i & -i;
}
int sum(int i)
{
    int s = 0;
    while(i > 0) s += bit[i], i -= i & -i;
    return s;
}
void build(int &rt, int l, int r)
{
    rt = ++tot;
    val[rt] = 0;
    if(l == r) return;
    int mid = (l + r) >> 1;
    build(lson[rt], l, mid);
    build(rson[rt], mid + 1, r);
}
void update(int pre, int &rt, int l, int r, int v, int x)
{
    rt = ++tot;
    lson[rt] = lson[pre], rson[rt] = rson[pre], val[rt] = val[pre] + x;
    if(l == r) return;
    int mid = (l + r) >> 1;
    if(v <= mid) update(lson[pre], lson[rt], l, mid, v, x);
    else update(rson[pre], rson[rt], mid + 1, r, v, x);
}
int query(int ss, int tt, int l, int r, int k)
{
    if(l == r) return l;
    int mid = (l + r) >> 1;
    int tmp = val[lson[tt]] - val[lson[ss]];
    if(k <= tmp) return query(lson[ss], lson[tt], l, mid, k);
    else return query(rson[ss], rson[tt], mid + 1, r, k - tmp);
}
int main()
{
    int m, a, b, c;
    while(~ scanf("%d", &m))
    {
        memset(num, 0, sizeof num);
        memset(bit, 0, sizeof bit);
        tot = cnt = 0;
        build(root[0], 1, n);
        int k = 1;
        for(int i = 1; i <= m; i++)
        {
            scanf("%d", &a);
            if(a == 0)
            {
                cnt++;//cnt计数器，统计当前容器中的元素个数
                scanf("%d", &b);
                num[b]++; //统计每一种数字的个数
                add(b, 1);
                update(root[k-1], root[k], 1, n, b, 1); //插入新元素新建节点
                k++;
            }
            else if(a == 1)
            {
                scanf("%d", &b);
                if(num[b] != 0)
                {
                    cnt--;
                    add(b, -1);
                    update(root[k-1], root[k], 1, n, b, -1); //删除元素新建节点
                    num[b]--;
                    k++;
                }
                else printf("No Elment!\n"); //数字不存在
            }
            else
            {
                scanf("%d%d", &b, &c);
                int res = sum(b);
                if(res + c > cnt) printf("Not Find!\n"); //当小于等于a的数字个数 + k > 容器中元素个数时，必定无解
                else printf("%d\n", query(root[0], root[k-1], 1, n, res + c));
            }
        }
    }
    return 0;
}