pku2761 Feed the dogs

没什么好说的,和2104差不多,写个SBT模板直接套

 

#include <iostream>
#include <algorithm>
using namespace std;

#define MAXN 100001

int a[MAXN],n,m;
int key[3*MAXN],lc[3*MAXN],rc[3*MAXN],sz[3*MAXN],root,cnt;

struct Query{
    int x,y,tag,k;
}query[50001];
int ans[50001];


///////////////////////////////

int search(int t,int v);//返回下标

void insert(int &t,int v);//修改了树的结构，故参数需引用

int deletion(int &t,int v);//返回下标

int pred(int t,int v);//返回下标

int succ(int t,int v);//返回下标

int select(int t,int k);//返回下标

int rank(int t,int v);

//////////////////////////////


void RightRotate(int &t){
    int k=lc[t];
    lc[t]=rc[k];
    rc[k]=t;
    sz[k]=sz[t];
    sz[t]=sz[lc[t]]+sz[rc[t]]+1;
    t=k;
}

void LeftRotate(int &t){
    int k=rc[t];
    rc[t]=lc[k];
    lc[k]=t;
    sz[k]=sz[t];
    sz[t]=sz[lc[t]]+sz[rc[t]]+1;
    t=k;
}

int search(int t,int v){
    if(!t || v==key[t])
        return t;
    if(v<key[t])
        return search(lc[t],v);
    else
        return search(rc[t],v);
}

void Maintain(int &t,bool RightDeeper){//注意右旋的条件是左子树存在，左旋的条件是右子树存在
    if(!RightDeeper){
        if(lc[t] && sz[lc[lc[t]]]>sz[rc[t]])
            RightRotate(t);
        else if(lc[t] && rc[lc[t]] && sz[rc[lc[t]]]>sz[rc[t]]){
            LeftRotate(lc[t]);
            RightRotate(t);
        }
        else
            return;
    }
    else{
        if(rc[t] && sz[rc[rc[t]]]>sz[lc[t]])
            LeftRotate(t);
        else if(rc[t] && lc[rc[t]] && sz[lc[rc[t]]]>sz[lc[t]]){
            RightRotate(rc[t]);
            LeftRotate(t);
        }
        else
            return;
    }
    Maintain(lc[t],false);
    Maintain(rc[t],true);
    Maintain(t,false);
    Maintain(t,true);
}

void insert(int &t,int v){
    if(!t){
        ++cnt;
        t=cnt;
        key[t]=v;
        lc[t]=0;
        rc[t]=0;
        sz[t]=1;
        return;
    }
    sz[t]++;
    if(v<key[t])
        insert(lc[t],v);
    else
        insert(rc[t],v);
    Maintain(t,v>=key[t]);
}

int deletion(int &t, int v){//如果树中没有一个这样的结点，删除搜索到的最后一个结点并返回其指针  
    if(!t)
        return 0;
    sz[t]--;
    if(v==key[t] || v<key[t]&&!lc[t] || v>key[t]&&!rc[t]){
        int del;
        if(!lc[t]||!rc[t]){
            del=t;
            //T=(T->lc?T->lc:T->rc);
            t=lc[t]+rc[t];
        }
        else{//t有两个儿子
            del=deletion(rc[t],v-1);//v==key[t]，记录t的后继并复制到t，删除后继
            key[t]=key[del];  //若有卫星数据也需复制
        }
        return del;
    }
    else return deletion(v<key[t]?lc[t]:rc[t],v);
}



int pred(int t,int v){
    if(!t)
        return 0;
    if(v<=key[t])
        return pred(rc[t],v);
    else{
        int p=pred(rc[t],v);
        return (p?p:t);
    }
}

int succ(int t,int v){
    if(!t)
        return 0;
    if(v>=key[t])
        return succ(rc[t],v);
    else{
        int s=succ(lc[t],v);
        return (s?s:t);
    }
}
 
int select(int t,int k){
    if(!t||k>sz[t])
        return 0;
    int r=(lc[t]?sz[lc[t]]:0)+1;
    if(k==r)
        return t;
    if(k<r)
        return select(lc[t],k);
    else
        return select(rc[t],k-r);
}


int rank(int t,int v){
    if(!t)
        return 0;
    if(v==key[t])
        return (lc[t]?sz[lc[t]]:0)+1;
    if(v<key[t])
        return rank(lc[t],v);
    else{
        int r=rank(rc[t],v);
        return r?r+(lc[t]?sz[lc[t]]:0)+1:0;
    }
}

class CP{
public:
    int operator()(Query a,Query b){
        if(a.x==b.x)
            return a.y<b.y;
        return a.x<b.x;
    }
};


int main(){
    int i,j;
    while(scanf("%d%d",&n,&m)!=EOF){
        
        for(i=1;i<=n;i++)
            scanf("%d",&a[i]);
        for(i=1;i<=m;i++){
            scanf("%d%d%d",&query[i].x,&query[i].y,&query[i].k);
            if(query[i].x>query[i].y){
                int temp=query[i].x;
                query[i].x=query[i].y;
                query[i].y=temp;
            }
            query[i].tag=i;
        }
        sort(query+1,query+m+1,CP());

        root=cnt=0;
        for(j=query[1].x;j<=query[1].y;j++)
            insert(root,a[j]);
        ans[query[1].tag]=key[select(root,query[1].k)];
        
        for(i=2;i<=m;i++){
            if(query[i-1].y<query[i].x || query[i-1].x<query[i].x && query[i].y<query[i-1].y){
                root=cnt=0;
                //for(j=query[i-1].x;j<=query[i-1].y;j++)
                //    deletion(root,a[j]);
                for(j=query[i].x;j<=query[i].y;j++)
                    insert(root,a[j]);
                
            }
            else{
                for(j=query[i-1].x;j<=query[i].x-1;j++)
                    deletion(root,a[j]);
                for(j=query[i-1].y+1;j<=query[i].y;j++)
                    insert(root,a[j]);
            }
            ans[query[i].tag]=key[select(root,query[i].k)];
            
        }
        for(i=1;i<=m;i++)
            printf("%d\n",ans[i]);
    }
    return 0;
}