AC日记——[SDOI2017]树点涂色 bzoj 4817

4817

思路：

　　跪烂大佬

代码：

#include <bits/stdc++.h>
using namespace std;
#define maxn 500005
struct TreeNodeType {
    int l,r,mid,dis,flag;
};
struct TreeNodeType tree[maxn<<];
int n,m,rev[maxn],f[maxn],ch[maxn][],Qes,deep[maxn],id[maxn];
int f_[maxn],top[maxn],tp,sta[maxn],size[maxn],lar[maxn];
int li[maxn],ri[maxn],cnt,E[maxn<<],V[maxn<<],head[maxn];
inline void in(int &now)
{
    char Cget=getchar();now=;
    while(Cget>''||Cget<'')Cget=getchar();
    while(Cget>=''&&Cget<='')
    {
        now=now*+Cget-'';
        Cget=getchar();
    }
}
void build(int now,int l,int r)
{
    tree[now].l=l,tree[now].r=r;
    if(l==r){tree[now].dis=deep[id[l]];return;}
    tree[now].mid=l+r>>;build(now<<,l,tree[now].mid);
    build(now<<|,tree[now].mid+,r);
    tree[now].dis=max(tree[now<<].dis,tree[now<<|].dis);
}
inline void down(int now)
{
    tree[now<<].dis+=tree[now].flag;
    tree[now<<].flag+=tree[now].flag;
    tree[now<<|].dis+=tree[now].flag;
    tree[now<<|].flag+=tree[now].flag;
    tree[now].flag=;
}
void add(int now,int l,int r,int x)
{
    if(tree[now].l>=l&&tree[now].r<=r)
    {
        tree[now].dis+=x,tree[now].flag+=x;
        return;
    }
    if(tree[now].flag!=) down(now);
    if(l<=tree[now].mid) add(now<<,l,r,x);
    if(r>tree[now].mid) add(now<<|,l,r,x);
    tree[now].dis=max(tree[now<<].dis,tree[now<<|].dis);
}
void query(int now,int l,int r)
{
    if(tree[now].l>=l&&tree[now].r<=r)
    {
        Qes=max(Qes,tree[now].dis);
        return;
    }
    if(tree[now].flag!=) down(now);
    if(l<=tree[now].mid) query(now<<,l,r);
    if(r>tree[now].mid) query(now<<|,l,r);
}
void downdata(int now)
{
    rev[now]^=,swap(ch[now][],ch[now][]);
    if(ch[now][]) rev[ch[now][]]^=;
    if(ch[now][]) rev[ch[now][]]^=;
}
bool isroot(int now)
{
    return (ch[f[now]][]!=now)&&(ch[f[now]][]!=now);
}
void rotate(int now)
{
    int fa=f[now],ffa=f[fa],l=(ch[fa][]==now),r=l^;
    if(!isroot(fa)) ch[ffa][ch[ffa][]==fa]=now;
    f[now]=ffa,f[fa]=now,ch[fa][l]=ch[now][r],ch[now][r]=fa;
    if(ch[fa][l]) f[ch[fa][l]]=fa;
}
void splay(int now)
{
    tp=,sta[tp]=now;int fa,ffa;
    for(int i=now;!isroot(i);i=f[i]) sta[++tp]=f[i];
    while(tp)
    {
        if(rev[sta[tp]]) downdata(sta[tp]);
        tp--;
    }
    while(!isroot(now))
    {
        fa=f[now],ffa=f[fa];
        if(!isroot(fa)) rotate(((ch[ffa][]==fa)^(ch[fa][]==now))?fa:now);
        rotate(now);
    }
}
int getson(int now)
{
    while(ch[now][]) now=ch[now][];
    return now;
}
void access(int now)
{
    int tmp;
    for(int i=;now;i=now,now=f[now])
    {
        splay(now);
        tmp=getson(ch[now][]);
        if(tmp) add(,li[tmp],ri[tmp],);
        ch[now][]=i,tmp=getson(i);
        if(tmp) add(,li[tmp],ri[tmp],-);
    }
}
void makeroot(int now)
{
    access(now),splay(now),rev[now]^=;
}
void link(int x,int y)
{
    makeroot(x),f[x]=y;
}
void edge_add(int u,int v)
{
    E[++cnt]=head[u],V[cnt]=v,head[u]=cnt;
    E[++cnt]=head[v],V[cnt]=u,head[v]=cnt;
}
void dfs1(int now,int fa)
{
    deep[now]=deep[fa]+,li[now]=++cnt;
    id[cnt]=now,size[now]=,f_[now]=fa;
    for(int i=head[now];i;i=E[i])
    {
        if(V[i]==fa) continue;
        dfs1(V[i],now),size[now]+=size[V[i]];
        if(size[lar[now]]<size[V[i]]) lar[now]=V[i];
    }
    ri[now]=cnt;
}
void dfs2(int now,int chain)
{
    top[now]=chain;
    if(f_[now]) f[now]=f_[now];
    if(lar[now]) dfs2(lar[now],chain);
    for(int i=head[now];i;i=E[i])
    {
        if(V[i]==lar[now]||V[i]==f_[now]) continue;
        dfs2(V[i],V[i]);
    }
}
int find(int x,int y)
{
    while(top[x]!=top[y])
    {
        if(deep[top[x]]<deep[top[y]]) y=f_[top[y]];
        else x=f_[top[x]];
    }
    return deep[x]<deep[y]?x:y;
}
int main()
{
    freopen("data.txt","r",stdin);
    in(n),in(m);int u,v,op;
    for(int i=;i<n;i++) in(u),in(v),edge_add(u,v);
    cnt=,dfs1(,),build(,,n),dfs2(,),makeroot();
    while(m--)
    {
        in(op);
        if(op==) in(u),makeroot(),access(u);
        if(op==)
        {
            in(u),in(v);
            int lca=find(u,v);
            Qes=,query(,li[u],li[u]),u=Qes;
            Qes=,query(,li[v],li[v]),v=Qes;
            if(lca) Qes=,query(,li[lca],li[lca]),lca=Qes;
            else lca=;
            printf("%d\n",u+v-lca*+);
        }
        if(op==) in(u),Qes=,query(,li[u],ri[u]),printf("%d\n",Qes);
    }
    return ;
}