社区讨论
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 条回复,欢迎继续交流。
正在加载回复...