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
, lson[N*20], rson[N*20], val[N*20];
int bit
, num
;
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;
}