Edison UVALive3488

传送门

题目大意

有一个0~n-1的序列,有m次操作,操作包含三个元素:pl,len,ti,表示这个操作进行ti次,每次将从pl+1开始的len个元素移到序列头部。
分析

看到题不难想到使用平衡树将需移动部分切出来挂到顶部。本题我使用fhqtreap来实现,但需要注意的是对于挪动部分使用的split不再以节点值为关键字切割而改用它在树中的排名作为关键字。具体实现详见代码。注意输出格式!!!!!!!!

代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<algorithm>
#include<cctype>
#include<cmath>
#include<cstdlib>
#include<queue>
#include<ctime>
#include<vector>
#include<set>
#include<map>
#include<stack>
using namespace std;
int root,son[1100000][2],a[1100000],rd[1100000],siz[1100000],cnt,n;
inline void up(int x){siz[x]=siz[son[x][0]]+siz[son[x][1]]+1;}
inline int bnode(int x){
      siz[++cnt]=1;
      a[cnt]=x;
      rd[cnt]=rand();
      return cnt;
}
inline int gk(int rt,int k){
      while(1){
        if(k<=siz[son[rt][0]])rt=son[rt][0];
          else if(k==siz[son[rt][0]]+1)return rt;
          else k-=siz[son[rt][0]]+1,rt=son[rt][1];    
      }
}
inline void spt1(int rt,int k,int &x,int &y){
      if(!rt)x=y=0;
      else {
          if(a[rt]<=k)x=rt,spt1(son[rt][1],k,son[rt][1],y);
            else y=rt,spt1(son[rt][0],k,x,son[rt][0]);
          up(rt);
      }
}
inline int mer(int x,int y){
      if(!x||!y)return x+y;
      if(rd[x]<rd[y]){
          son[x][1]=mer(son[x][1],y);
          up(x);
          return x;
      }else {
          son[y][0]=mer(x,son[y][0]);
          up(y);
          return y;
      }
}
inline void spt(int sum,int rt,int k,int &x,int &y){
      if(!rt)x=y=0;
      else {
          if(sum+siz[son[rt][0]]+1<=k){
          x=rt,spt(sum+siz[son[rt][0]]+1,
          son[rt][1],k,son[rt][1],y);
        }else y=rt,spt(sum,son[rt][0],k,x,son[rt][0]);
          up(rt);
      }
}
inline void init(){
      root=cnt=0;
      memset(rd,0,sizeof(rd));
      memset(a,0,sizeof(a));
      memset(son,0,sizeof(son));
      memset(siz,0,sizeof(siz));
}
inline long long getans(int n){
      long long ans=0;
      int i;
      for(i=1;i<=n;i+=2)
        ans+=(long long)a[gk(root,i)];
      return ans;
}
inline void work(int _){
      int n,m,i,j,x,y,z,w,pl,len,ti;
      scanf("%d%d",&n,&m);
      for(i=0;i<n;i++){
          spt1(root,i,x,y);
          root=mer(mer(x,bnode(i)),y);
      }
      for(i=1;i<=m;i++){
          scanf("%d%d%d",&pl,&len,&ti);
          for(j=1;j<=ti;j++){
            spt(0,root,pl,x,y),spt(0,y,len,z,w);
            root=mer(z,mer(x,w));
        }
      }
      printf("%lld\n",getans(n));      
}
int main(){
      srand(time(0)+20021023);
      int t;
      scanf("%d",&t);
      for(int _=1;_<=t;_++){
          printf("Case %d:\n",_);
          init();
          work(_);
          if(_!=t)puts("");
      }
      return 0;
}
posted @ 2018-07-30 15:22  水题收割者  阅读(136)  评论(0编辑  收藏  举报