社区讨论

60ptsWA求条玄关

P10289[GESP样题 八级] 小杨的旅游参与者 1已保存回复 1

讨论操作

快速查看讨论及其快照的属性,并进行相关操作。

当前回复
1 条
当前快照
1 份
快照标识符
@mhjhpyov
此快照首次捕获于
2025/11/04 02:45
4 个月前
此快照最后确认于
2025/11/04 02:45
4 个月前
查看原帖
CPP
#include<bits/stdc++.h>
using namespace std;
int n,k,q,dep[200001],dis[200001],anc[200001][20];//dis表示当前节点里传送门的距离,dep表示里1的距离,anc表示第i个节点的第2^j级祖先
vector<int>g[200001];//图
queue<int>qu;//bfs
void dfs(int x,int last){
	for(int i=1;i<=18;i++)
		anc[x][i]=anc[anc[x][i-1]][i-1];
	for(int y:g[x]){
		if(y==last) continue;
		dep[y]=dep[x]+1;
		anc[y][0]=x;
		dfs(y,x);
	}
}
int lca(int x,int y){
	if(dep[x]<dep[y]) swap(x,y);
	for(int i=18;i>=0;i--)
		if(dep[anc[x][i]]>=dep[y]) x=anc[x][i];
	if(x==y) return x;
	for(int i=18;i>=0;i--)
		if(anc[x][i]!=anc[y][i]) x=anc[x][i],y=anc[y][i];
	return anc[x][0];
}
int main(){
	memset(dis,0x3f3f3f3f,sizeof dis);
	cin>>n>>k>>q;
	for(int i=1,x,y;i<n;i++){
		cin>>x>>y;
		g[x].push_back(y),g[y].push_back(x);
	}
	for(int i=1,x;i<=k;i++){
		cin>>x;
		qu.push(x);
		dis[x]=0;
	}
	while(!qu.empty()){//当前节点里传送门的距离
		int x=qu.front();qu.pop();
		for(int y:g[x])
			if(dis[y]==0x3f3f3ff3) dis[y]=dis[x]+1,qu.push(y);
	}
	dep[1]=1;
	dfs(1,0);//倍增求LCA
	while(q--){
		int x,y;
		cin>>x>>y;
		cout<<min(dep[x]+dep[y]-2*dep[lca(x,y)],dis[x]+dis[y])<<'\n';
	}
	return 0;
}

回复

1 条回复,欢迎继续交流。

正在加载回复...