题意
给出一棵有根树,支持更改父节点、子树加权、询问根到某一点路径上的权值和。
N≤1e5。
题解
Euler Tour Tree的模板。
听jcy讲的时候感觉比LCT简单 于是写了一发 调了两个晚自修…。
ETT是用splay维护这棵树的欧拉序,就是形如”1,2,-2,3,-3,-1”这样的 每个点正编号在第一次访问到时出现,负编号在回溯时出现 的序列(并不会准确描述
子树加就是把那一段子树转出来。
换父亲的话,把整个子树(x和-x中间那一段)平移到-y前面就好。
一个点到根的路径,就是1~x的权值和。
正编号的位置是这个点的权值,负编号的位置是这个点权值的相反数 推标记的时候注意一下
代码
为什么跑得这么慢啊…
#include<bits/stdc++.h>
#define N 200005
using namespace std;
int n,m,cnt,f[N],c[N][2],dfn[N],des[N],
to[N],hd[N],lk[N],val[N],ll[N],rr[N],siz[N];
long long sum[N],a[N],laz[N];
void dfs(int k)
{
dfn[ll[k]=++cnt]=k;
for(int i=lk[k];i;i=hd[i])
dfs(to[i]);
dfn[rr[k]=++cnt]=-k;
}
inline void update(int k)
{
sum[k]=sum[c[k][0]]+sum[c[k][1]]+a[k];
siz[k]=siz[c[k][0]]+siz[c[k][1]]+des[k];
}
int build(int l,int r)
{
int mid=l+r>>1;
if(l<mid)
f[c[mid][0]=build(l,mid-1)]=mid;
if(mid<r)
f[c[mid][1]=build(mid+1,r)]=mid;
a[mid]=dfn[mid]>0?val[dfn[mid]]:-val[-dfn[mid]];
des[mid]=dfn[mid]>0?1:-1;
update(mid);
return mid;
}
inline void add(int k,int v)
{laz[k]+=v,a[k]+=des[k]*v,sum[k]+=(long long)siz[k]*v;}
void pd(int k)
{
if(f[k])pd(f[k]);
if(laz[k])
add(c[k][0],laz[k]),add(c[k][1],laz[k]),laz[k]=0;
}
inline void rotate(int x,bool mode)
{
int y=f[x];
if(f[y])
c[f[y]][c[f[y]][1]==y]=x;
f[x]=f[y],f[c[x][mode]]=y;
c[y][mode^1]=c[x][mode],c[x][mode]=y,f[y]=x;
update(y),update(x);
}
inline void splay(int x,int k)
{
pd(x);
bool t1,t2;int y;
while((y=f[x])^k)
{
t1=c[y][0]==x,t2=c[f[y]][0]==y;
if(f[y]==k){rotate(x,t1);return;}
(t1^t2)?(rotate(x,t1),rotate(x,t2)):
(rotate(y,t2),rotate(x,t1));
}
}
inline int findl(int x)
{
splay(x,0);
x=c[x][0];
while(c[x][1])x=c[x][1];
return x;
}
inline int findr(int x)
{
splay(x,0);
x=c[x][1];
while(c[x][0])x=c[x][0];
return x;
}
char op[2];
int x,y,z,tp;
int main()
{
scanf("%d",&n);
for(int i=2;i<=n;i++)
scanf("%d",&x),to[++cnt]=i,
hd[cnt]=lk[x],lk[x]=cnt;
cnt=1,dfs(1);
for(int i=1;i<=n;i++)
scanf("%d",val+i);
build(1,cnt+1);
scanf("%d",&m);
while(m--)
{
scanf("%s%d",op,&x);
if(op[0]=='Q')
{
splay(ll[x],0);
printf("%lld\n",sum[c[ll[x]][0]]+a[ll[x]]);
}
if(op[0]=='C')
{
tp=findl(ll[x]),z=findr(rr[x]);
splay(tp,0),splay(z,tp);
y=c[z][0];
c[z][0]=0;
update(z),update(tp);
x=y;
scanf("%d",&y);
splay(z=findl(rr[y]),0),splay(rr[y],z);
f[x]=rr[y],c[rr[y]][0]=x;
update(rr[y]),update(z);
}
if(op[0]=='F')
{
scanf("%d",&y);
tp=findl(ll[x]),z=findr(rr[x]);
splay(tp,0),splay(z,tp);
add(c[z][0],y);
}
}
}