HDU 3727 Jewel 主席树简单题

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

题意：有一个空链，现有如下几种操作：

Insert x 把数x插入链中（从右端插入）
Query_1 s t k 查询链上区间[s, t]内的第k小值
Query_2 x 查询x在链上是第几小
Query_3 k 查询整条链上的第k小值

思路：首先把所有操作存下来，把要插入的数据离散化，然后求k小值很简单，不多说，至于查询x是第几小，用一个树状数组维护即可

总结：此题中注意树状数组添加值时，如果一直更新到题目所给的n，那么树状数组的长度应该是大于n即135000，但是因为只有100000个插入数据，所以我只开了100010，哇了好几次，小心！！！

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

typedef long long ll;
const int N = 100010;
struct node
{
int f, v, u, k;
}g[N*2];
int root
, lson[N*20], rson[N*20], val[N*20];
int a
, bit[N*2];
int n, tot, cas;
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 l, int r, int &rt)
{
rt = ++tot;
val[rt] = 0;
if(l == r) return;
int mid = (l + r) >> 1;
build(l, mid, lson[rt]);
build(mid + 1, r, rson[rt]);
}
void update(int pre, int &rt, int l, int r, int v)
{
rt = ++tot;
lson[rt] = lson[pre], rson[rt] = rson[pre], val[rt] = val[pre] + 1;
if(l == r) return;
int mid = (l + r) >> 1;
if(v <= mid) update(lson[pre], lson[rt], l, mid, v);
else update(rson[pre], rson[rt], mid + 1, r, v);
}
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()
{
while(~ scanf("%d", &n))
{
char s[20];
int k = 0;
for(int i = 1; i <= n; i++)
{
scanf("%s", s);
if(s[0] == 'I')
{
g[i].f = 0;
scanf("%d", &g[i].k);
a[++k] = g[i].k;
}
else
{
g[i].f = s[6] - '0';
if(g[i].f == 1) scanf("%d%d%d", &g[i].v, &g[i].u, &g[i].k);
else scanf("%d", &g[i].k);
}
}
sort(a+1, a+1+k);
tot = 0;
build(1, k, root[0]);
memset(bit, 0, sizeof bit);
int cnt = 0;
ll sum1 = 0, sum2 = 0, sum3 = 0;
for(int i = 1; i <= n; i++)
{
if(g[i].f == 0)
{
g[i].k = lower_bound(a+1, a+1+k, g[i].k) - a;
update(root[cnt], root[cnt+1], 1, k, g[i].k);
add(g[i].k, 1);
cnt++;
}
else if(g[i].f == 1)
{
int res = query(root[g[i].v-1], root[g[i].u], 1, k, g[i].k);
sum1 += a[res];
}
else if(g[i].f == 2)
{
g[i].k = lower_bound(a+1, a+1+k, g[i].k) - a;
sum2 += sum(g[i].k);
}
else
{
int res = query(root[0], root[cnt], 1, k, g[i].k);
sum3 += a[res];
}
}
printf("Case %d:\n%lld\n%lld\n%lld\n", ++cas, sum1, sum2, sum3);
}
return 0;
}