BZOJ 3786 星系探索 (splay+dfs序)

题目大意：给你一棵树，支持一下三种操作

1.获取某节点到根节点的路径上所有节点的权值和

2.更换某棵子树的父亲

3.某子树内所有节点的权值都增加一个值w

当时想到了splay维护dfs序，查完题解发现思路是对的，然后我就写了足足6个小时才A

st[x]代表入栈时间，ed[x]代表出栈时间

对于第一个操作，每个树上节点在splay中都有两个位置，分别对应入栈出栈序，然后把入栈的点权*1，出栈点权*-1，就可以避免其它子树干扰了

接着查询到根节点的路径权值和呢，splay把1~st[x]整个序列都扔到一个子树里，求一下子树权值和即可

第二个操作，可以找一下st[x]和ed[x]对应splay中节点的前驱和后继，把它们都搞到一个splay子树里，在把整棵树接到要求的父节点即可

第三个操作，和第二个操作类似，找前驱后继然后在子树打标记即可，此外，还要维护splay子树内的入栈节点数-出栈节点数来直接得到splay子树的权值和。

注意，一定要在所有能传标记的地方都传标记，不论是从下往上splay，还是从上到下找前驱or后继，而如果不在找前驱后继的过程中下传标记，上层的标记无法被传到下层，导致结果错误！就这个错误我调了2个小时！

 #include <cstdio>
 #include <cstring>
 #include <algorithm>
 #define N 200100
 #define il inline
 #define ll long long
 #define root d[0].ch[1]
 using namespace std;

 char q[];
 int n,m,cte,tot;
 int st[N],ed[N],head[N],sq[N];
 ll w[N];
 struct EDGE{int to,nxt;}edge[N];
 struct SPLAY{int fa,ch[],ss,fl;ll val,sum,tag;}d[N];
 il int idf(int x){return d[d[x].fa].ch[]==x?:;}
 il void con(int x,int ff,int p){d[ff].ch[p]=x;d[x].fa=ff;}
 il void pushup(int x){
     d[x].sum=d[d[x].ch[]].sum+d[d[x].ch[]].sum+d[x].val;
     d[x].ss=d[d[x].ch[]].ss+d[d[x].ch[]].ss+d[x].fl;
 }
 il void pushdown(int x){
     if(!d[x].tag) return;
     int ls=d[x].ch[],rs=d[x].ch[];
     d[ls].tag+=d[x].tag,d[rs].tag+=d[x].tag;
     d[ls].val+=d[x].tag*(ll)d[ls].fl,d[rs].val+=d[x].tag*(ll)d[rs].fl;
     d[ls].sum+=d[x].tag*(ll)d[ls].ss,d[rs].sum+=d[x].tag*(ll)d[rs].ss;
     d[x].tag=;
 }
 il void rot(int x){
     int y=d[x].fa;int ff=d[y].fa;int px=idf(x);int py=idf(y);
     pushdown(y),pushdown(x);
     con(d[x].ch[px^],y,px),con(y,x,px^),con(x,ff,py);
     pushup(y),pushup(x);
 }
 il void splay(int x,int to){
     to=d[to].fa;
     while(d[x].fa!=to){
         int y=d[x].fa;
         if(d[y].fa==to) rot(x);
         else if(idf(x)==idf(y)) rot(y),rot(x);
         else rot(x),rot(x);}
 }
 int gc(){
     int rett=,fh=;char p=getchar();
     while(p<''||p>''){if(p=='-')fh=-;p=getchar();}
     while(p>=''&&p<=''){rett=(rett<<)+(rett<<)+p-'';p=getchar();}
     return rett*fh;
 }
 void ae(int u,int v){
     cte++;edge[cte].to=v;
     edge[cte].nxt=head[u];head[u]=cte;
 }
 void dfs1(int u,int ff)
 {
     st[u]=++tot,sq[tot]=u,d[tot].val=w[u],d[tot].fl=;
     for(int j=head[u];j!=-;j=edge[j].nxt){
         int v=edge[j].to;
         if(v==ff) continue;
         dfs1(v,u);}
     ed[u]=++tot,sq[tot]=u,d[tot].val=-w[u],d[tot].fl=-;
 }
 int build(int l,int r)
 {
     if(l>r) return ;
     int mid=(l+r)>>;
     con(build(l,mid-),mid,);
     con(build(mid+,r),mid,);
     pushup(mid);
     return mid;
 }
 int lower(int x)
 {
     pushdown(x),x=d[x].ch[];
     while(x){
         pushdown(x);
         if(d[x].ch[]) x=d[x].ch[];
         else break;}
     return x;
 }
 int upper(int x)
 {
     pushdown(x),x=d[x].ch[];
     while(x){
         pushdown(x);
         if(d[x].ch[]) x=d[x].ch[];
         else break;}
     return x;
 }
 ll query(int x)
 {
     splay(st[],root);
     splay(st[x],d[root].ch[]);
     int lw=upper(st[x]);
     splay(lw,d[root].ch[]);
     return d[d[lw].ch[]].sum;
 }
 void add(int x,ll ww)
 {
     int trl,trr,rt;
     splay(st[x],root);
     splay(ed[x],d[root].ch[]);
     trl=lower(st[x]);
     splay(trl,root);
     trr=upper(ed[x]);
     splay(trr,d[root].ch[]);
     rt=d[trr].ch[];
     d[rt].val+=ww*(ll)d[rt].fl;
     d[rt].sum+=ww*(ll)d[rt].ss;
     d[rt].tag+=ww;
 }
 void mov(int x,int y)
 {
     int trl,trr,frt,rtt;
     splay(st[x],root);
     splay(ed[x],d[root].ch[]);
     trl=lower(st[x]);
     splay(trl,root);
     trr=upper(ed[x]);
     splay(trr,d[root].ch[]);
     pushdown(trl),pushdown(trr);
     rtt=d[trr].ch[];
     d[trr].ch[]=,d[rtt].fa=;
     pushup(trr),pushup(trl);

     splay(st[y],root);
     splay(ed[y],d[root].ch[]);
     trl=lower(ed[y]);
     if(trl) con(rtt,trl,);
     else con(rtt,ed[y],);
     splay(rtt,root);
 }
 void print(int x)
 {
     if(d[x].ch[]) print(d[x].ch[]);
     printf("%d ",sq[x]);
     if(d[x].ch[]) print(d[x].ch[]);
 }

 int main()
 {
     scanf("%d",&n);
     int x,y;
     memset(head,-,sizeof(head));
     for(int i=;i<=n;i++)
         x=gc(),ae(x,i),ae(i,x);
     for(int i=;i<=n;i++)
         w[i]=gc();
     st[]=++tot,dfs1(,-),ed[]=++tot;
     root=build(,tot);
     scanf("%d",&m);
     for(int i=;i<=m;i++){
         scanf("%s",q);
         if(q[]=='Q'){
             x=gc();
             printf("%lld\n",query(x));
         }else if(q[]=='F'){
             x=gc(),y=gc();
             add(x,(ll)y);
         }else{
             x=gc(),y=gc();
             mov(x,y);
         }
     }
     /*for(int i=1;i<=n;i++)
         printf("%lld %lld %lld\n",query(i),d[st[i]].val,d[ed[i]].val);*/
     return ;
 }