Codeforces Round #199 (Div. 2) E - Xenia and Tree(lca+分块(定期重构))

题目

n(n<=1e5)个点的树，树上的点有颜色

初始情况，除了根节点1是红色以外，其他点都是蓝色

m(m<=1e5)次操作，操作分两种

1 u 把u染成红色

2 u 询问离u最近的红色点的距离 若u为红色输出0

思路来源

https://blog.csdn.net/u013451221/article/details/47132707?utm_source=blogxgwz3

题解

倍增预处理lca，定期重构bfs，也叫分块

对于1操作，在不满sz的情况下，先存到vector，到vector的阈值sz，再多源bfs更新到所有点的距离

对于2操作，直接暴力更新，因为vector内的点的个数没到阈值sz

由于lca的复杂度是log，sz取sqrt(m/logm)比较好吧，似乎取sqrt(m)也无所谓

代码

#include<bits/stdc++.h> 
using namespace std;
typedef long long ll;
typedef pair<int,int> P;
#define pb push_back
const int N=1e5+10;
const int INF=0x3f3f3f3f;

int n,m,sz,op,u,v,ans;
int dep[N],f[N][18];
int d[N];
vector<int>E[N];
vector<int>now;

void dfs(int u,int fa)
{
	for(int i=0;i<E[u].size();++i)
	{
		int v=E[u][i];
		if(v==fa)continue;
		dep[v]=dep[u]+1; //必须前序 否则传参 
		dfs(v,u);
		f[v][0]=u;
	}
}
void init()
{
	for(int i=1;i<18;++i)
	{
		for(int j=1;j<=n;++j)
		f[j][i]=f[f[j][i-1]][i-1];
	}
}
int lca(int x,int y)
{
	if(dep[x]<dep[y])swap(x,y);
	int d=dep[x]-dep[y];
	for(int i=17;i>=0;--i)
	if(d>>i&1)x=f[x][i];
	if(x==y)return x;
	for(int i=17;i>=0;--i)
	{
		if(f[x][i]!=f[y][i])
		x=f[x][i],y=f[y][i];
	}
	return f[x][0];
}
void bfs()
{
	priority_queue<P,vector<P>,greater<P> >q;
	for(int i=0;i<sz;++i)
	{
		int v=now[i];
		d[v]=0;q.push(P(0,v));
	}
	while(!q.empty())
	{
		int dis=q.top().first,u=q.top().second;
		q.pop();
		if(d[u]<dis)continue;
		int len=E[u].size();
		for(int i=0;i<len;++i)
		{
			int v=E[u][i];
			if(d[v]>d[u]+1)
			{
				d[v]=d[u]+1;
				q.push(P(d[v],v));
			}
		}	
	} 
}
int main()
{
	scanf("%d%d",&n,&m);
	for(int i=1;i<n;++i)
	{
		scanf("%d%d",&u,&v);
		E[v].pb(u);E[u].pb(v);
	}
	dfs(1,-1);init();
	memset(d,INF,sizeof d);
	now.pb(1);
	sz=sqrt(1.0*m/log(m));
	for(int i=1;i<=m;++i)
	{
		scanf("%d%d",&op,&u);
		if(op==1)
		{
			now.pb(u);
			if(now.size()==sz)
			{
				bfs();
				now.clear();
			}
		}
		else if(op==2)
		{
			ans=d[u]; 
			for(int i=0;i<now.size();++i)
			{
				v=now[i];
				ans=min(ans,dep[u]+dep[v]-2*dep[lca(u,v)]);
			}
			printf("%d\n",ans);
		}
	}
    return 0;
}