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 ;
}