CF342E Xenia and Tree（分块暴力）

每个点记一个dis[x],表示离他最近的红点的距离。每次把一个点染红就记下来这个点，当超过$\sqrt n$个时就暴力bfs重新搜一遍，并清空数组。每次询问x，就是若干还没更新的红点与x的距离和dis[x]取min。复杂度$O(n\sqrt n)$

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
#include <queue>
using namespace std;
#define inf 0x3f3f3f3f
#define ll long long
#define N 100010
inline char gc(){
    static char buf[1<<16],*S,*T;
    if(S==T){T=(S=buf)+fread(buf,1,1<<16,stdin);if(T==S) return EOF;}
    return *S++;
}
inline int read(){
    int x=0,f=1;char ch=gc();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=gc();}
    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=gc();
    return x*f;
}
int n,m,h[N],num=0,dis[N],dep[N],pos[N],dfn=0,mn[N<<1][20],Log[N<<1],fa[N];
int nn,tot=0,red[N];
struct edge{
    int to,next;
}data[N<<1];
inline void dfs(int x){
    mn[++dfn][0]=x;pos[x]=dfn;
    for(int i=h[x];i;i=data[i].next){
        int y=data[i].to;if(y==fa[x]) continue;
        fa[y]=x;dep[y]=dep[x]+1;dis[y]=dep[y];dfs(y);mn[++dfn][0]=x;
    }
}
inline int Min(int x,int y){return dep[x]<dep[y]?x:y;}
inline void inirmq(){
    Log[0]=-1;
    for(int i=1;i<=n<<1;++i) Log[i]=Log[i>>1]+1;
    for(int i=1;i<=Log[n*2-1];++i)
        for(int j=1;j<n<<1;++j){
            if(j+(1<<i-1)>n*2-1) break;
            mn[j][i]=Min(mn[j][i-1],mn[j+(1<<i-1)][i-1]);
        }
}
inline int lca(int x,int y){
    x=pos[x],y=pos[y];
    if(x>y) swap(x,y);int d=Log[y-x+1];
    return Min(mn[x][d],mn[y-(1<<d)+1][d]);
}
inline int caldis(int x,int y){return dep[x]+dep[y]-2*dep[lca(x,y)];}
inline void bfs(){
    queue<int>q;
    for(int i=1;i<=tot;++i) q.push(red[i]),dis[red[i]]=0;
    while(!q.empty()){
        int x=q.front();q.pop();
        for(int i=h[x];i;i=data[i].next){
            int y=data[i].to;
            if(dis[x]+1<dis[y]) dis[y]=dis[x]+1,q.push(y);
        }
    }
}
int main(){
//  freopen("a.in","r",stdin);
    n=read();m=read();nn=sqrt(n);
    for(int i=1;i<n;++i){
        int x=read(),y=read();
        data[++num].to=y;data[num].next=h[x];h[x]=num;
        data[++num].to=x;data[num].next=h[y];h[y]=num;
    }dfs(1);inirmq();
    while(m--){
        int op=read(),x=read();
        if(op==1){
            red[++tot]=x;if(tot==nn) bfs(),tot=0;continue;
        }int res=dis[x];
        for(int i=1;i<=tot;++i){
            int y=red[i];res=min(res,caldis(x,y));
        }printf("%d\n",res);
    }return 0;
}