树上算法总结

背景

LittlePrincess做了一道又一道的题，学了一些树上算法，于是决定总结一波

树剖

很久之前学习的算法，主要解决树上两点之间的修改和权值查询功能
主要思想：用数据结构维护重链来达到$logn$修改及查询
代码

#include<bits/stdc++.h>
using namespace std;
#define rig register
#define MN 500010
long long c1[MN],w[MN],c2[MN];
int h[MN],son[MN],wson[MN],fa[MN],
top[MN],dep[MN],lk[MN],zs[MN],T,tot,n,m,r,p;bool vis[MN],b[MN];
int lb(int x){return x&-x;}
void addd(int pos,long long data)
{
    for(rig int i=pos;i<=n;i+=lb(i)) 
    c1[i]+=data,c2[i]+=pos*data;//,c1[i]%=p,c2[i]%=p;
}
long long  gsum(int pos)
{
    long long ans=0;for(rig int i=pos;i;i-=lb(i)) 
    ans+=(pos+1)*c1[i]-c2[i]/*,ans%=p;*/;return ans;
}
struct lll{
    int to,ne;
}a[MN*2];
void link(int x,int y){a[++T].to=y,a[T].ne=h[x],h[x]=T;}
int dfs(int u)
{
    int i=h[u];int M=-2100000000,wz;
    while(i)
    {
        if(!vis[a[i].to])
        {
            fa[a[i].to]=u;dep[a[i].to]=dep[u]+1;
            vis[a[i].to]=1;int ha=dfs(a[i].to);son[u]+=ha;
            if(ha>M) M=ha,wson[u]=a[i].to;
        }i=a[i].ne;
    }son[u]++;
    return son[u];
}
void DFS(int u)
{
    lk[u]=++tot;b[u]=1;top[wson[u]]=top[u],DFS(wson[u]);
    int i=h[u];while(i){if(!b[a[i].to]) DFS(a[i].to);i=a[i].ne;}zs[u]=tot;
}
void do1(int x,int y,long long z)
{
    if(top[x]==top[y]) if(dep[x]<dep[y]) {addd(lk[x],z),addd(lk[y]+1,-z);return;}
                else {addd(lk[y],z),addd(lk[x]+1,-z);return ;}
    else if(dep[top[x]]>dep[top[y]]) 
    addd(lk[top[x]],z),addd(lk[x]+1,-z),do1(fa[top[x]],y,z);
    else addd(lk[top[y]],z),addd(lk[y]+1,-z),do1(x,fa[top[y]],z);
}
long long do2(int x,int y)
{
    long long ans=0;
    if(top[x]==top[y]) if(dep[x]<dep[y]) {return ans=gsum(lk[y])-gsum(lk[x]-1);}
                else {return ans=gsum(lk[x])-gsum(lk[y]-1);}
    else if(dep[top[x]]>dep[top[y]]) 
    ans+=gsum(lk[x])-gsum(lk[top[x]]-1),ans+=do2(fa[top[x]],y);
    else ans+=gsum(lk[y])-gsum(lk[top[y]]-1)+do2(x,fa[top[y]]);return ans;
}
void do3(int x,long long z){addd(lk[x],z),addd(zs[x]+1,-z);}
long long do4(int x){return gsum(zs[x])-gsum(lk[x]-1);}
int main()
{
    ios::sync_with_stdio(0);
    cin>>n>>m>>r>>p;
    vis[r]=1;b[0]=1;fa[r]=r;
    for(rig int i=1;i<=n;i++) cin>>w[i],top[i]=i;
    for(rig int i=1,_,__;i<n;i++) cin>>_>>__,link(_,__),link(__,_);
    int _o_=dfs(r); DFS(r);
    for(rig int i=1;i<=n;i++) 
    addd(lk[i],w[i]),addd(lk[i]+1,-w[i]);
    for(rig int i=1,A;i<=m;i++)
    {
        long long z,x,y;cin>>A;
        if(A==1)cin>>x>>y>>z,do1(x,y,z);
        if(A==2) cin>>x>>y,cout<<do2(x,y)%p<<endl;
        if(A==3)    cin>>x>>y,do3(x,y);
        if(A==4) cin>>x,cout<<do4(x)%p<<endl;
    }
    return 0;
}

LCT

该数据结构主要用来维护link cut 问题，即动态树，但也有一些玄学用处，如SDOI2017树点染色。
主要思想：利用splay维护深度来达到不修改主树而通过辅助树可以查询主树的功能
代码

#include<bits/stdc++.h>
using namespace std;
#define so(i,j,k,l) for(int i=j;i<=k;i+=l)
#define MN 300030
#define lc ch[x][0]
#define rc ch[x][1]
#define I void
#define R register int
int ch[MN][2],n,m,v[MN],s[MN],st[MN],f[MN],opt,x,y;
bool tag[MN];
bool nroot(int x)
{return ch[f[x]][0]==x||ch[f[x]][1]==x;}
void pushup(int x)
{
    s[x]=s[ch[x][0]]^s[ch[x][1]]^v[x];
}
void pushr(int x)
{
    swap(ch[x][0],ch[x][1]);
    tag[x]^=1;
}
void pushdown(int x)
{
    if(tag[x])
    {
        if(ch[x][0]) pushr(ch[x][0]);
        if(ch[x][1]) pushr(ch[x][1]);
        tag[x]=0;
    }
}
void rotate(int x)
{
    int old=f[x],oldf=f[old],which=ch[old][1]==x,w=ch[x][!which];
    if(nroot(old)) ch[oldf][ch[oldf][1]==old]=x;
    ch[x][which^1]=old;ch[old][which]=w;
    if(w) f[w]=old;
    f[old]=x;f[x]=oldf;
    pushup(old);
}
void splay(int x)
{
    int y=x,z=0;
    st[++z]=y;
    while(nroot(y)) st[++z]=y=f[y];
    while(z) pushdown(st[z--]);
    while(nroot(x))
    {
        int fa=f[x],ffa=f[fa];
        if(nroot(fa))
        rotate((ch[fa][0]==x)^(ch[ffa][0]==fa)?x:fa);
        rotate(x);
    }
    pushup(x);
}
void access(int x)
{
    for(int y=0;x;x=f[y=x])
    {
        splay(x),ch[x][1]=y,pushup(x);
    }
}
void mkrt(int x)
{
    access(x);splay(x);
    pushr(x);
}
int findrt(int x)
{
    access(x),splay(x);
    while(ch[x][0]) pushdown(x),x=ch[x][0];
    return x;
}
void split(int x,int y)
{
    mkrt(x),access(y),splay(y);
}
void link(int x,int y)
{
    mkrt(x);
    if(findrt(x)!=findrt(y)) f[x]=y;    
}
void cut(int x,int y)
{
    split(x,y);
    if(findrt(x)==findrt(y)&&f[x]==y&&!ch[x][1])
    {
        f[x]=ch[y][0]=0;pushup(y);
    }
}

int main()
{
    scanf("%d%d",&n,&m);
    so(i,1,n,1) scanf("%d",&v[i]);
    while(m--)
    {
        scanf("%d%d%d",&opt,&x,&y);
        switch(opt)
        {
            case 0:split(x,y);printf("%d\n",s[y]);break;
            case 1:link(x,y);break;
            case 2:cut(x,y);break;
            case 3:splay(x);v[x]=y;
        }
    }
    return 0;
}

点分治

同样是解决树上两点之间的权值问题，但是与树剖不同的是树剖有$m$次查询每次给你$(u,v)$，而点分治是解决形如：树上任意两点之间距离为k的个数的问题的
主要思想：利用重心的特点来完成$nlon$的分治~

#include<bits/stdc++.h>
using namespace std;
#define inf 2100000000
#define MN 10010
struct lll{
    int to,w,ne;
}a[2*MN];
int n,m,sum[10000010],tot,S,MX,cnt,toot,h[MN],root,dis[MN],sim[MN],mxson[MN],k,x,y,z;
bool vis[MN]; 
void link(int x,int y,int z)
{
    a[++tot].to=y,a[tot].w=z,a[tot].ne=h[x],h[x]=tot;
}
void getroot(int u,int fa)
{
    sim[u]=1;mxson[u]=0;
    for(int i=h[u];i;i=a[i].ne)
    {
        int v=a[i].to;
        if(vis[v]||v==fa) continue;
        getroot(v,u);
        sim[u]+=sim[v];
        mxson[u]=max(mxson[u],sim[v]);
    }
    mxson[u]=max(mxson[u],S-sim[u]);
    if(mxson[u]<MX) MX=mxson[u],root=u;
}
void ask(int u,int fa,int len)
{
    dis[++cnt]=len;
    for(int i=h[u];i;i=a[i].ne)
    {
        int v=a[i].to;
        if(vis[v]||v==fa) continue;
        ask(v,u,len+a[i].w);
    }
} 
void solve(int u,int len,int emm)
{
    cnt=0;
    ask(u,0,len);
    for(int i=1;i<=cnt;i++)
    for(int j=i+1;j<=cnt;j++)
    sum[dis[i]+dis[j]]+=emm;
}
void doit(int u)
{
    vis[u]=1;
    solve(u,0,1);
    for(int i=h[u];i;i=a[i].ne)
    {
        int v=a[i].to;
        if(vis[v]) continue;
        solve(v,a[i].w,-1);
        MX=inf,root=0,S=sim[v];
        getroot(v,0);
        doit(root);
    }
    return ;
}
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<n;i++) scanf("%d%d%d",&x,&y,&z),link(x,y,z),link(y,x,z);
    S=n,MX=inf;
    getroot(1,0);
    doit(root);
    for(int i=1;i<=m;i++) 
    {scanf("%d",&k);if(sum[k]) printf("AYE\n");else printf("NAY\n");}
}