洛谷 P3203 [HNOI2010]弹飞绵羊 || bzoj2002

看来这个lct板子的确没什么问题

好像还可以分块做

 #include<cstdio>
 #include<algorithm>
 using namespace std;
 typedef long long LL;
 namespace LCT
 {
 struct Node
 {
     Node *ch[],*fa;
     bool rev;
     LL sz;
     void upd()
     {
         sz=(ch[]?ch[]->sz:)+(ch[]?ch[]->sz:)+;
     }
     void pd()
     {
         if(rev)
         {
             swap(ch[],ch[]);
             if(ch[])    ch[]->rev^=;
             if(ch[])    ch[]->rev^=;
             rev=;
         }
     }
 }nodes[];
 LL mem;
 Node *getnode()
 {
     return nodes+(mem++);
 }
 bool isroot(Node *x)
 {
     return (!x->fa)||((x->fa->ch[]!=x)&&(x->fa->ch[]!=x));
 }
 bool gson(Node *o)    {return o==o->fa->ch[];}//获得是父亲的左儿子(返回0)还是右儿子(1)，要求保证存在父亲
 void rotate(Node *o,bool d)
 //在o子树中执行d=0左旋，d=1右旋，在旋转前不标记下传，并将o父节点的对应子节点由o变为需要值，要求保证存在子树(!d)
 {
     Node *k=o->ch[!d];if(!isroot(o))    o->fa->ch[gson(o)]=k;//注意这一句修改o父节点的要写在前面，曾经出过错调了一会
     o->ch[!d]=k->ch[d];k->ch[d]=o;
     o->upd();k->upd();
     k->fa=o->fa;o->fa=k;if(o->ch[!d])    o->ch[!d]->fa=o;
 }
 Node *st[];LL top;
 void solvetag(Node *o)
 {
     while(!isroot(o))    st[++top]=o,o=o->fa;
     st[++top]=o;
     while(top)    st[top--]->pd();
 }
 void splay(Node *o)
 {
     solvetag(o);
     Node *fa,*fafa;bool d1,d2;
     while(!isroot(o))
     {
         fa=o->fa;d1=(o==fa->ch[]);
         if(isroot(fa))    rotate(fa,d1);
         else
         {
             fafa=o->fa->fa;d2=(fa==fafa->ch[]);//要保证fa不是root之后才能获取这两个值，曾错过
             if(d1==d2)    rotate(fafa,d1),rotate(fa,d1);//zig-zig，两次相同方向的单旋，先把父亲转上去，再把自己转上去
             else    rotate(fa,d1),rotate(fafa,d2);//zig-zag，两次相反方向的单旋，连续两次把自己转上去
         }
     }
 }
 void access(Node *o)
 {
     for(Node *lst=NULL;o;lst=o,o=o->fa)
     {
         splay(o);//此处不pushdown是由于splay中保证进行过了
         o->ch[]=lst;o->upd();//注意upd
     }
 }
 Node *gtop(Node *o)
 {
     access(o);splay(o);
     for(;o->ch[];o=o->ch[],o->pd());//此处不在开始前pushdown(o)是由于splay中保证进行过了
     splay(o);return o;//听说这里不splay一下也很难卡掉
 }
 void mtop(Node *o)    {access(o);splay(o);o->rev^=;}
 void link(Node *x,Node *y)
 {
     if(gtop(x)==gtop(y))    return;
     mtop(y);y->fa=x;
 }
 void cut(Node *x,Node *y)
 {
     mtop(x);access(y);splay(y);
     if(y->ch[]!=x||x->ch[])    return;//如果x、y之间直接有边，那么上面一行的操作之后应当是x与y在单独一棵splay中，那么一定保证y左子节点是x且x没有右子节点
     x->fa=y->ch[]=NULL;//注意，改的是x的父亲和y的子节点(虽然x的确是树的根，但是此时在splay上是y的子节点，不能搞混)
     y->upd();//注意
 }
 LL query(Node *x,Node *y)
 {
     mtop(x);access(y);splay(y);
     //if(gtop(y)!=x)    return 0;//此题保证x与y连通，不需要
     return y->sz;
 }
 }
 LCT::Node *nd[];
 LL n,m;char tmp[];LL a[];
 int main()
 {
     LL i,x,y,idx;
     scanf("%lld",&n);
     for(i=;i<=n;i++)
     {
         nd[i]=LCT::getnode();
         nd[i]->sz=;
     }
     for(i=;i<n;i++)
     {
         scanf("%lld",&a[i]);
         LCT::link(nd[i],nd[min(i+a[i],n)]);
     }
     scanf("%lld",&m);
     for(i=;i<=m;i++)
     {
         scanf("%lld%lld",&idx,&x);
         if(idx==)
         {
             printf("%lld\n",LCT::query(nd[x],nd[n])-);
         }
         else
         {
             scanf("%lld",&y);
             LCT::cut(nd[x],nd[min(x+a[x],n)]);
             a[x]=y;
             LCT::link(nd[x],nd[min(x+a[x],n)]);
         }
     }
     return ;
 }