(CodeForces) E. Xenia and Tree (lca+分塊+最短路)

傳送門

題意:給定一棵樹,一開始只有1爲紅,其他點爲藍。兩種操作:1,把一個點染成紅點。2,詢問一個點到最近紅點的距離。

:數據量只有1e5,n根號n*log應該是可以的,主要是這個染色後,我們不能每一次染色後都去跑一次最短路(bfs就可以了),所以我們當修改數目達到根號n時再去進行一次最短路,查詢是如果有點是未更新狀態,我們可以通過lca來求得兩點的距離,這樣就可以保證複雜度是可行的。

#include<bits/stdc++.h>
#define il inline
#define pb push_back
#define ms(_data,v) memset(_data,v,sizeof(_data))
#define SZ(a) int((a).size())
using namespace std;
typedef long long ll;
const ll inf=0x3f3f3f3f;
const int N=1e5+5;
const int maxbit=20;
//il int Add(ll &x,ll y) {return x=x+y>=mod?x+y-mod:x+y;}
//il int Mul(ll &x,ll y) {return x=x*y>=mod?x*y%mod:x*y;}
int n,m,base;
int dep[N],fa[N][maxbit],lg[N],dis[N];//dis:到紅點最近的距離 
vector<int> G[N];
bool vis[N];
il void init(){
	lg[0]=-1,base=sqrt(n)+1,vis[1]=1;
	for(int i=1;i<N;++i) lg[i]=lg[i>>1]+1;
}
il void dfs(int np,int f){
	dep[np]=dep[f]+1,fa[np][0]=f,dis[np]=dep[np]-1;
	for(int i=1;i<=lg[dep[np]]+1;++i) fa[np][i]=fa[fa[np][i-1]][i-1];
	for(auto to:G[np]){
		if(to!=f) dfs(to,np);
	}
}
il int lca(int u,int v){
	if(dep[u]<dep[v]) swap(u,v);
	while(dep[u]!=dep[v])	u=fa[u][lg[dep[u]-dep[v]]];
	if(u==v) return u;
	for(int i=lg[dep[u]];i>=0;--i){
		if(fa[u][i]!=fa[v][i]) u=fa[u][i],v=fa[v][i];
	}
	return fa[u][0];
}
il int dist(int u,int v){
	return dep[u]+dep[v]-2*dep[lca(u,v)];
}
int q[N],cnt=0;
queue<int> qe;
il void solve(){
	int x;
	while(!qe.empty()){
		x=qe.front(),qe.pop();
		for(auto to:G[x]){
			if(dis[to]>dis[x]+1){
				dis[to]=dis[x]+1;
				qe.push(to);
			}
		}
	}
}
il void update(int x){
	if(vis[x]) return;
	vis[x]=1,dis[x]=0,q[++cnt]=x,qe.push(x);
	if(cnt==base){
		solve();
		cnt=0;
	}
}
il int query(int x){
	int ans=dis[x];
	for(int i=1;i<=cnt;++i) ans=min(ans,dist(q[i],x));
	return ans;
}
int main(){
	std::ios::sync_with_stdio(0);cin.tie(0);
	cin>>n>>m;
	init();
	for(int i=1,x,y;i<=n-1;++i){
		cin>>x>>y;
		G[x].pb(y),G[y].pb(x);
	}
	dfs(1,0);
	for(int i=1,op,x;i<=m;++i){
		cin>>op>>x;
		if(op==1) update(x);
		else cout<<query(x)<<endl;
	}
	return 0;
}



 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章