CF342E Xenia and Tree

首先思考一下暴力怎么写：（有以下两种很明显的方法）

1.把每个不断得到的红点都存储下来，对于后面的每个询问，与每个红点通过lca求得点距，取最小值。

2.每得到一个红点，就对于整棵树做一次bfs，更新树中每个点到最近红点的距离。

我们发现，如果红点数量为n/2个，询问数量为n/2个，那么两种暴力思想都会被卡。

那么我们就把红点分个块来处理吧。

假设红点数量和为S。

将红点分为sqrt（S）个块，每个块内sqrt（S）个红点。

当每得到一整个块个数的红点时（即sqrt(S)），就对整棵树做一次bfs。复杂度：sqrt(S)*n

对于每次查询，我们可得到：对于前面若干个整块的红点，已更新最优答案。所以现在，只需要将剩余的最多sqrt(S)个红点，与查询点做lca枚举最小值即可。复杂度：sqrt(S)*n

总复杂度即为：2sqrt(S)*n

即2sqrt(n)*n

将查询（即暴力1）或更新（即暴力2），由n均摊为sqrt(n)，降一下次数，即可得到正确的复杂度。

#include <bits/stdc++.h>
using namespace std;
const int N=1e5+5;
int n,m,u,v,opt,x,size,now,ans;
int p[N][21],d[N],minn[N],vec[N];
bool col[N],vis[N];
int cnt,head[N];
struct edge{int next,to;}e[N<<1];

inline void add(int u,int v)
{
	cnt++;
	e[cnt].next=head[u];
	e[cnt].to=v;
	head[u]=cnt;
}

void dfs(int u,int fa)
{
	for (register int i=1; (1<<i)<=d[u]; ++i) p[u][i]=p[p[u][i-1]][i-1];
	for (register int i=head[u]; i; i=e[i].next)
	if (e[i].to!=fa)
	{
		d[e[i].to]=d[u]+1;
		p[e[i].to][0]=u;
		minn[e[i].to]=minn[u]+1;
		dfs(e[i].to,u);
	}
}

queue<int>q;
inline void bfs()
{
	for (register int i=1; i<=n; ++i) vis[i]=false;
	for (register int i=1; i<=n; ++i) if (col[i]) minn[i]=0,q.push(i),vis[i]=true;
	while (q.size())
	{
		int u=q.front(); q.pop();
		for (register int i=head[u]; i; i=e[i].next) 
		if (!vis[e[i].to]) 
		{
			minn[e[i].to]=minn[u]+1;
			q.push(e[i].to); vis[e[i].to]=true;
		}
	}
}

inline int lca(int a,int b)
{
	if (d[a]>d[b]) swap(a,b);
	for (register int i=20; i>=0; --i) if (d[b]-(1<<i)>=d[a]) b=p[b][i];
	if (a==b) return a;
	for (register int i=20; i>=0; --i) 
	if (p[a][i]==p[b][i]) continue;
	else a=p[a][i],b=p[b][i];
	return p[a][0];
}	

inline int dis(int a,int b)
{
	return d[a]+d[b]-2*d[lca(a,b)];	
}

int main(){
	scanf("%d%d",&n,&m);
	for (register int i=1; i<n; ++i) scanf("%d%d",&u,&v),add(u,v),add(v,u);
	d[1]=1; minn[1]=0; dfs(1,0);

	col[1]=true;
	size=sqrt(n)+1;
	while (m--)
	{
		scanf("%d%d",&opt,&x);
		if (opt==1)
		{
			vec[++now]=x; col[x]=true;
			if (now==size)
			{
				bfs();
				now=0;
			}	
		}
		else
		{
			ans=minn[x];
			for (register int i=1; i<=now; ++i) ans=min(ans,dis(x,vec[i]));
			printf("%d\n",ans);
		}	
	}
return 0;
}