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)>>1const 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;}