luogu P3703 [SDOI2017]树点涂色

luogu

看到操作1,应该学过lct的都知道这个等价于access操作,所以可以考虑用lct维护,相同颜色的一条链就用lct上的一棵splay维护,每次操作1就\(access(x)\)

要回答操作2和操作3,都要求出某点到根的颜色段数.考虑\(access(x)\)过程中会改变一些点的颜色段数,当某条边变成虚边后,那么对应儿子子树内所有点的颜色数都加\(1\),当某条边变成实边后,对应儿子子树内所有点的颜色数都减\(1\),这个不理解可以手动模拟access.然后操作二答案为\(a_x+a_y-2*a_{lca}+1\),就是两条从lca往下的链答案+lca的贡献.操作三要询问子树,但是一个点子树在\(dfs\)序上是连续区间,所以线段树维护每个点以及子树内答案即可

//补档
#include<bits/stdc++.h>
#define LL long long
#define il inline
#define re register
#define ldb long double

using namespace std;
const int N=100000+10;
il LL rd()
{
    LL x=0,w=1;char ch;
    while(ch<'0'||ch>'9') {if(ch=='-') w=-1;ch=getchar();}
    while(ch>='0'&&ch<='9') {x=(x<<3)+(x<<1)+(ch^48);ch=getchar();}
    return x*w;
}
int to[N<<1],nt[N<<1],hd[N],tot=1;
il void add(int x,int y)
{
    ++tot,to[tot]=y,nt[tot]=hd[x],hd[x]=tot;
    ++tot,to[tot]=x,nt[tot]=hd[y],hd[y]=tot;
}
int n,m,ff[N],de[N],sz[N],hson[N],top[N],dfn[N],id[N],tt;
int fa[N],ch[N][2];
void dfs1(int x)
{
    sz[x]=1;
    for(int i=hd[x];i;i=nt[i])
    {
        int y=to[i];
        if(y==ff[x]) continue;
        ff[y]=fa[y]=x,de[y]=de[x]+1,dfs1(y),sz[x]+=sz[y];
        if(sz[hson[x]]<sz[y]) hson[x]=y;
    }
}
void dfs2(int x,int ntp)
{
    dfn[x]=++tt,id[tt]=x,top[x]=ntp;
    if(hson[x]) dfs2(hson[x],ntp);
    for(int i=hd[x];i;i=nt[i])
    {
        int y=to[i];
        if(y==ff[x]||y==hson[x]) continue;
        dfs2(y,y);
    }
}
il int glca(int x,int y)
{
    while(top[x]!=top[y])
    {
        if(de[top[x]]<de[top[y]]) swap(x,y);
        x=ff[top[x]];
    }
    return de[x]<de[y]?x:y;
}
int ma[N<<2],lz[N<<2];
#define lc (o<<1)
#define rc ((o<<1)|1)
#define mid ((l+r)>>1)
il void psup(int o){ma[o]=max(ma[lc],ma[rc]);}
il void upd(int o,int x){ma[o]+=x,lz[o]+=x;}
il void psdn(int o)
{
    if(lz[o]) upd(lc,lz[o]),upd(rc,lz[o]),lz[o]=0;
}
void bui(int o,int l,int r)
{
    if(l==r) {ma[o]=de[id[l]];return;}
    bui(lc,l,mid),bui(rc,mid+1,r);
    psup(o);
}
void modif(int o,int l,int r,int ll,int rr,int x)
{
    if(ll<=l&&r<=rr)
    {
        upd(o,x);
        return;
    }
    psdn(o);
    if(ll<=mid) modif(lc,l,mid,ll,rr,x);
    if(rr>mid) modif(rc,mid+1,r,ll,rr,x);
    psup(o);
}
int quer(int o,int l,int r,int ll,int rr)
{
    if(ll<=l&&r<=rr) return ma[o];
    int an=0;
    psdn(o);
    if(ll<=mid) an=max(an,quer(lc,l,mid,ll,rr));
    if(rr>mid) an=max(an,quer(rc,mid+1,r,ll,rr));
    psup(o);
    return an;
}
il bool nrt(int x){return ch[fa[x]][0]==x||ch[fa[x]][1]==x;}
il void rot(int x)
{
    int y=fa[x],z=fa[y],yy=ch[y][1]==x,w=ch[x][!yy];
    if(nrt(y)) ch[z][ch[z][1]==y]=x;
    ch[y][yy]=w,ch[x][!yy]=y;
    if(w) fa[w]=y;
    fa[y]=x,fa[x]=z;
}
il void spl(int x)
{
    int y,z;
    while(nrt(x))
    {
        y=fa[x],z=fa[y];
        if(nrt(y)) ((ch[y][1]==x)^(ch[z][1]==y))?rot(x):rot(y);
        rot(x);
    }
}
il int findrt(int x)
{
    while(ch[x][0]) x=ch[x][0];
    return x;
}
il void acs(int x)
{
    for(int y=0,z;x;y=x,x=fa[x])
    {
        spl(x);
        if(ch[x][1]) z=findrt(ch[x][1]),modif(1,1,n,dfn[z],dfn[z]+sz[z]-1,1);
        if(y) z=findrt(y),modif(1,1,n,dfn[z],dfn[z]+sz[z]-1,-1);
        ch[x][1]=y;
    }
}

int main()
{
    n=rd(),m=rd();
    for(int i=1;i<n;++i) add(rd(),rd());
    de[1]=1,dfs1(1),dfs2(1,1),bui(1,1,n);
    while(m--)
    {
        int op=rd();
        if(op==1) acs(rd());
        else if(op==2)
        {
            int x=rd(),y=rd(),lca=glca(x,y);
            printf("%d\n",quer(1,1,n,dfn[x],dfn[x])+quer(1,1,n,dfn[y],dfn[y])-(quer(1,1,n,dfn[lca],dfn[lca])<<1)+1);
        }
        else
        {
            int x=rd();
            printf("%d\n",quer(1,1,n,dfn[x],dfn[x]+sz[x]-1));
        }
    }
    return 0;
}

猜你喜欢

转载自www.cnblogs.com/smyjr/p/11493004.html