题目大意:给你一棵树,支持一下三种操作
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[10];
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[2],ss,fl;ll val,sum,tag;}d[N];
il int idf(int x){return d[d[x].fa].ch[0]==x?0:1;}
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[0]].sum+d[d[x].ch[1]].sum+d[x].val;
d[x].ss=d[d[x].ch[0]].ss+d[d[x].ch[1]].ss+d[x].fl;
}
il void pushdown(int x){
if(!d[x].tag) return;
int ls=d[x].ch[0],rs=d[x].ch[1];
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=0;
}
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^1],y,px),con(y,x,px^1),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=0,fh=1;char p=getchar();
while(p<'0'||p>'9'){if(p=='-')fh=-1;p=getchar();}
while(p>='0'&&p<='9'){rett=(rett<<3)+(rett<<1)+p-'0';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=1;
for(int j=head[u];j!=-1;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=-1;
}
int build(int l,int r)
{
if(l>r) return 0;
int mid=(l+r)>>1;
con(build(l,mid-1),mid,0);
con(build(mid+1,r),mid,1);
pushup(mid);
return mid;
}
int lower(int x)
{
pushdown(x),x=d[x].ch[0];
while(x){
pushdown(x);
if(d[x].ch[1]) x=d[x].ch[1];
else break;}
return x;
}
int upper(int x)
{
pushdown(x),x=d[x].ch[1];
while(x){
pushdown(x);
if(d[x].ch[0]) x=d[x].ch[0];
else break;}
return x;
}
ll query(int x)
{
splay(st[0],root);
splay(st[x],d[root].ch[1]);
int lw=upper(st[x]);
splay(lw,d[root].ch[1]);
return d[d[lw].ch[0]].sum;
}
void add(int x,ll ww)
{
int trl,trr,rt;
splay(st[x],root);
splay(ed[x],d[root].ch[1]);
trl=lower(st[x]);
splay(trl,root);
trr=upper(ed[x]);
splay(trr,d[root].ch[1]);
rt=d[trr].ch[0];
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[1]);
trl=lower(st[x]);
splay(trl,root);
trr=upper(ed[x]);
splay(trr,d[root].ch[1]);
pushdown(trl),pushdown(trr);
rtt=d[trr].ch[0];
d[trr].ch[0]=0,d[rtt].fa=0;
pushup(trr),pushup(trl);
splay(st[y],root);
splay(ed[y],d[root].ch[1]);
trl=lower(ed[y]);
if(trl) con(rtt,trl,1);
else con(rtt,ed[y],0);
splay(rtt,root);
}
void print(int x)
{
if(d[x].ch[0]) print(d[x].ch[0]);
printf("%d ",sq[x]);
if(d[x].ch[1]) print(d[x].ch[1]);
}
int main()
{
scanf("%d",&n);
int x,y;
memset(head,-1,sizeof(head));
for(int i=2;i<=n;i++)
x=gc(),ae(x,i),ae(i,x);
for(int i=1;i<=n;i++)
w[i]=gc();
st[0]=++tot,dfs1(1,-1),ed[0]=++tot;
root=build(1,tot);
scanf("%d",&m);
for(int i=1;i<=m;i++){
scanf("%s",q);
if(q[0]=='Q'){
x=gc();
printf("%lld\n",query(x));
}else if(q[0]=='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 0;
}