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