bzoj4817: [Sdoi2017]树点涂色【LCT+线段树】

Description

Bob有一棵n个点的有根树，其中1号点是根节点。Bob在每个点上涂了颜色，并且每个点上的颜色不同。定义一条路
径的权值是：这条路径上的点（包括起点和终点）共有多少种不同的颜色。Bob可能会进行这几种操作：
1 x:
把点x到根节点的路径上所有的点染上一种没有用过的新颜色。
2 x y:
求x到y的路径的权值。
3 x y:
在以x为根的子树中选择一个点，使得这个点到根节点的路径权值最大，求最大权值。
Bob一共会进行m次操作

Input

第一行两个数n,m。
接下来n-1行，每行两个数a,b，表示a与b之间有一条边。
接下来m行，表示操作，格式见题目描述
1<=n,m<=100000

Output

每当出现2,3操作，输出一行。
如果是2操作，输出一个数表示路径的权值
如果是3操作，输出一个数表示权值的最大值

Sample Input

5 6
1 2
2 3
3 4
3 5
2 4 5
3 3
1 4
2 4 5
1 5
2 4 5

Sample Output

3
4
2
2

解题思路：

初始时每个点答案就是它的深度。
先考虑操作二，发现一条链的权值为val[x]+val[y]-2*val[lca]+1,可以分lca和x,y都不同色或和其中一个同色讨论一下得到该结论。
至于操作三，用线段树就能dfs序即可。
最难的操作一让人想到LCT的access操作，画图分析发现一个点的答案就是它到根节点所经过的虚边条数+1，相当于实边连的是同种颜色的一条链，虚边连接了这几条链，所以我们直接在access切换轻重边的时候把原实儿子所在splay权值+1,新实儿子所在splay权值-1即可。不过为了配合线段树，应该找到splay的最浅点，也就是x在原树的真儿子，然后直接在线段树上修改。

#include<bits/stdc++.h>
using namespace std;
int getint()
{
    int i=0,f=1;char c;
    for(c=getchar();c!='-'&&(c<'0'||c>'9');c=getchar());
    if(c=='-')f=-1,c=getchar();
    for(;c>='0'&&c<='9';c=getchar())i=(i<<3)+(i<<1)+c-'0';
    return i*f;
}
void print(int x)
{
    if(x<0)putchar('-'),x=-x;
    if(x>9)print(x/10);
    putchar(x%10+'0');
}
const int N=100005;
int n,m;
int idx,in[N],out[N],id[N],dep[N],fa[N][20];
int tot,first[N],nxt[N<<1],to[N<<1];
int mx[N<<2],tag[N<<2];
inline void add(int x,int y)
{
    nxt[++tot]=first[x],first[x]=tot,to[tot]=y;
}
void dfs(int u)
{
    id[in[u]=++idx]=u;
    for(int i=1;i<20;i++)fa[u][i]=fa[fa[u][i-1]][i-1];
    for(int e=first[u];e;e=nxt[e])
    {
        int v=to[e];
        if(v==fa[u][0])continue;
        fa[v][0]=u,dep[v]=dep[u]+1;
        dfs(v);
    }
    out[u]=idx;
}
void pushdown(int k)
{
    mx[k<<1]+=tag[k],tag[k<<1]+=tag[k];
    mx[k<<1|1]+=tag[k],tag[k<<1|1]+=tag[k];
    tag[k]=0;
}
void build(int k,int l,int r)
{
    if(l==r){mx[k]=dep[id[l]];return;}
    int mid=l+r>>1;
    build(k<<1,l,mid),build(k<<1|1,mid+1,r);
    mx[k]=max(mx[k<<1],mx[k<<1|1]);
}
void modify(int k,int l,int r,int x,int y,int v)
{
    if(x<=l&&r<=y){mx[k]+=v,tag[k]+=v;return;}
    if(tag[k])pushdown(k);
    int mid=l+r>>1;
    if(x<=mid)modify(k<<1,l,mid,x,y,v);
    if(y>mid)modify(k<<1|1,mid+1,r,x,y,v);
    mx[k]=max(mx[k<<1],mx[k<<1|1]);
}
int query(int k,int l,int r,int x,int y)
{
    if(x==l&&r==y)return mx[k];
    if(tag[k])pushdown(k);
    int mid=l+r>>1;
    if(y<=mid)return query(k<<1,l,mid,x,y);
    else if(x>mid)return query(k<<1|1,mid+1,r,x,y);
    else return max(query(k<<1,l,mid,x,mid),query(k<<1|1,mid+1,r,mid+1,y));
}
int lca(int x,int y)
{
    if(dep[x]<dep[y])swap(x,y);
    int det=dep[x]-dep[y];
    for(int i=0;i<20;i++)if(det>>i&1)x=fa[x][i];
    for(int i=19;i>=0;i--)if(fa[x][i]!=fa[y][i])x=fa[x][i],y=fa[y][i];
    return x==y?x:fa[x][0];
}
struct LCT
{
    int fa[N],son[N][2];
    inline int which(int x){return son[fa[x]][1]==x;}
    inline bool rt(int x){return son[fa[x]][0]!=x&&son[fa[x]][1]!=x;}
    inline void rotate(int x)
    {
        int y=fa[x],z=fa[y],t=which(x);
        if(!rt(y))son[z][which(y)]=x;
        fa[x]=z,fa[y]=x;
        son[y][t]=son[x][t^1],son[x][t^1]=y;
        if(son[y][t])fa[son[y][t]]=y;
    }
    inline void splay(int x)
    {
        while(!rt(x))
        {
            if(!rt(fa[x]))which(x)==which(fa[x])?rotate(fa[x]):rotate(x);
            rotate(x);
        }
    }
    inline int get(int x)
    {
        while(son[x][0])x=son[x][0];
        return x;
    }
    inline void access(int x)
    {
        for(int y=0,z;x;y=x,x=fa[x])
        {
            splay(x);
            z=get(son[x][1]);
            if(z)modify(1,1,n,in[z],out[z],1);
            z=get(y);
            if(z)modify(1,1,n,in[z],out[z],-1);
            son[x][1]=y;if(y)fa[y]=x;
        }
    }
}lct;
int main()
{
    //freopen("lx.in","r",stdin);
    int op,x,y,t,ans;
    n=getint(),m=getint();
    for(int i=1;i<n;i++)
    {
        x=getint(),y=getint();
        add(x,y),add(y,x);
    }
    dep[1]=1,dfs(1);build(1,1,n);
    for(int i=1;i<=n;i++)lct.fa[i]=fa[i][0];
    while(m--)
    {
        op=getint();
        if(op==1)x=getint(),lct.access(x);
        else if(op==2)
        {
            x=getint(),y=getint(),t=lca(x,y),ans=0;
            ans+=query(1,1,n,in[x],in[x]);
            ans+=query(1,1,n,in[y],in[y]);
            ans-=query(1,1,n,in[t],in[t])*2;
            print(++ans),putchar('\n');
        }
        else x=getint(),print(query(1,1,n,in[x],out[x])),putchar('\n');
    }
    return 0;
}