bzoj3786: 星系探索 //ETT

bzoj3786: 星系探索


题意

给出一棵有根树,支持更改父节点、子树加权、询问根到某一点路径上的权值和。
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);
        }
    }
}
  • 2
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值