Codeforces Round #199 (Div. 2) E

题目链接：https://codeforces.com/problemset/problem/342/E

题意：给你一个树，起初根结点(1号结点）为红色，其余结点为蓝色，每次有两种操作。

1、把蓝色结点涂成红色

2、求给定结点v与树上最近的结点之间的距离，每条边权值都为1

分析：题解很巧妙，把m次操作中1操作分成块，每次对于1操作次数如果到了sqrt(m)，就用bfs更新已有的红色点i到其余蓝色点的最短距离_min[i]，然后弹出该块的红色结点，继续存下一个块的红色结点，每次询问答案是ans=min(_min[u],dist[u,v])

Ac code:

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=1e5+10;
const int INF=1e9;
int head[maxn];
int sze[maxn],fa[maxn],son[maxn],dfn[maxn];
int rk[maxn],dep[maxn],top[maxn],dist[maxn],f[maxn][21];
int _min[maxn];
vector<int>rednode;
struct Edge{
    int v,nxt;
}edge[maxn<<1];
int tot,cnt;
void init(int n)
{
    rednode.push_back(1);
    tot=cnt=0;
    memset(head,-1,sizeof head);
    dep[1]=1;
    dist[1]=0;
    for(int i=0;i<=n;i++) _min[i]=INF;
}
void addedge(int u,int v)
{
    edge[tot].v=v;
    edge[tot].nxt=head[u];
    head[u]=tot++;
}
void dfs1(int u,int faa)
{
    //dfn[u]=++cnt;
    sze[u]=1;
    for(int i=head[u];~i;i=edge[i].nxt){
        int v=edge[i].v;
        if(v==faa) continue;
        dep[v]=dep[u]+1;
        dist[v]=dist[u]+1;
        f[v][0]=u;
        for(int j=1;j<=20;j++)
            f[v][j]=f[f[v][j-1]][j-1];
        dfs1(v,u);
    }
}
int lca(int x, int y) {
    if(dep[x]>dep[y]) swap(x,y);
    for(int i=20;i>=0;--i)
        if(dep[f[y][i]]>=dep[x]) y=f[y][i];
    if(x==y) return x;
    for(int i=20;i>=0;--i)
        if(f[y][i]!=f[x][i]) x=f[x][i],y=f[y][i];
    return f[x][0];
}
void bfs()
{
    queue<int>q;
    for(int i=0;i<rednode.size();i++)
    {
        q.push(rednode[i]);
        _min[rednode[i]]=0;
    }
    while(!q.empty()){
        int u=q.front();
        q.pop();
        for(int i=head[u];~i;i=edge[i].nxt){
            int v=edge[i].v;
            if(v==fa[u]) continue;
            if(_min[v]>_min[u]+1){
                _min[v]=_min[u]+1;
                q.push(v);
            }
        }
    }
}
int main()
{
    int n,m,u,v;
    scanf("%d%d",&n,&m);
    init(n);
    for(int i=1;i<n;i++){
        scanf("%d%d",&u,&v);
        addedge(u,v);
        addedge(v,u);
    }
    dfs1(1,-1);
    //dfs2(1,1);
    int op;
    int block=sqrt(m)+1,cc=0;
    while(m--){
        scanf("%d%d",&op,&u);
        if(op==1){
            cc++;
            rednode.push_back(u);
            if(cc==block){
                cc=0;
                bfs();
                rednode.clear();
            }
        }
        else{
            int ans=_min[u];
            for(int i=0;i<rednode.size();i++)
            {
                int v=rednode[i];
                ans=min(ans,dist[u]+dist[v]-2*dist[lca(u,v)]);
            }
            printf("%d\n",ans);
        }
    }
    return 0;
}