HDU3727 Jewel(主席树)

传送门:http://acm.hdu.edu.cn/showproblem.php?pid=3727
这题还是挺需要细心的，算是主席树一个挺好的练手题了。
题意：输入n，再输入n个操作，操作有四种
Insert X：插入x到序列末尾
query1 L R X：在当前序列中的[l,r]区间找第x小的数。
query2 X：在当前序列中，输出X是第几小的数。
query3 X：找到当前序列中第X小的数是几。
然后输出的是3种query的和。

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
#include <cstdlib>
#include <cctype>
#include <string>
#include <iostream>
#include <vector>
#include <map>
#include <set>
#include <queue>
#include <ctime>
using namespace std;
typedef long long ll;
typedef pair<int,int> pii;
#define pb push_back
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
#define calm (l+r)>>1
const int INF=1e9+7;

const int maxn=100010;
struct Data{
   int t;
   int l,r,x;
}input[220000];
struct node{
    int l,r,sum;
}tree[maxn*20];
int n,tot,rt[maxn];
vector<int> v;
inline int getid(int x){
    return lower_bound(v.begin(),v.end(),x)-v.begin()+1;
}
void readinput(){
    char op[10];
    for(int i=0;i<n;i++){
        scanf("%s",op);
        if(op[0]=='I'){
            int x;scanf("%d",&x);
            input[i].t=0;
            input[i].x=x;
            v.pb(x);
        }
        else if(op[6]=='1'){
            int a,b,c;scanf("%d%d%d",&a,&b,&c);
            input[i].t=1;
            input[i].l=a;input[i].r=b;input[i].x=c;
        }
        else if(op[6]=='2'){
            int x;scanf("%d",&x);
            input[i].t=2;
            input[i].x=x;
        }
        else{
            int x;scanf("%d",&x);
            input[i].t=3;
            input[i].x=x;
        }
    }
}
void build(int l,int r,int &x){
    x=++tot;
    tree[x].sum=0;
    if(l==r)return;
    int m=calm;
    build(l,m,tree[x].l);
    build(m+1,r,tree[x].r);
}
void update(int l,int r,int &x,int y,int k){
    x=++tot;
    tree[x]=tree[y];tree[x].sum++;
    if(l==r)return;
    int m=calm;
    if(k<=m)update(l,m,tree[x].l,tree[y].l,k);
    else update(m+1,r,tree[x].r,tree[y].r,k);
}
int query1(int l,int r,int x,int y,int k){
    if(l==r)return l;
    int m=calm;
    int sum=tree[tree[y].l].sum-tree[tree[x].l].sum;
    if(k<=sum)return query1(l,m,tree[x].l,tree[y].l,k);
    else return query1(m+1,r,tree[x].r,tree[y].r,k-sum);
}
int query2(int l,int r,int x,int k){
    //printf("%d %d %d\n",l,r,k);
    if(l==r)return 1;
    int m=calm;
    if(k<=m)return query2(l,m,tree[x].l,k);
    else{
        int sum=tree[tree[x].l].sum;
        return sum+query2(m+1,r,tree[x].r,k);
    }
}
int query3(int l,int r,int x,int k){
    if(l==r)return l;
    int m=calm;
    int sum=tree[tree[x].l].sum;
    if(sum>=k)return query3(l,m,tree[x].l,k);
    else return query3(m+1,r,tree[x].r,k-sum);
}
int main(){
    //freopen("D://input.txt","r",stdin);
    int kase=1;
    while(scanf("%d",&n)!=EOF){
        v.clear();tot=0;
        ll ans1=0,ans2=0,ans3=0;
        readinput();
        sort(v.begin(),v.end());v.erase(unique(v.begin(),v.end()),v.end());
        int cnt=v.size();
        build(1,cnt,rt[0]);
        int now=1;
        for(int i=0;i<n;i++){
            if(input[i].t==0){
                update(1,cnt,rt[now],rt[now-1],getid(input[i].x));
                now++;
            }
            else if(input[i].t==1){
                int l=input[i].l,r=input[i].r,x=input[i].x;
                ans1+=v[query1(1,cnt,rt[l-1],rt[r],x)-1];
            }
            else if(input[i].t==2){
                ans2+=query2(1,cnt,rt[now-1],getid(input[i].x));
            }
            else if(input[i].t==3){
                ans3+=v[query3(1,cnt,rt[now-1],input[i].x)-1];
            }
        }
        printf("Case %d:\n",kase++);
        printf("%I64d\n%I64d\n%I64d\n",ans1,ans2,ans3);
    }
    return 0;
}