对着板子看了若干遍还是不知道为什么不行
查看原帖
对着板子看了若干遍还是不知道为什么不行
164840
zhaowangji楼主2020/7/22 14:54

dep是深度,fa和fat都是父亲,fro是现在所处点,to是通过边可到达的点

#include<bits/stdc++.h>
using namespace std;
int n,m,s;
int fa[500007][20],dep[500007];
vector<int> e[500007];

int lca(int x,int y){
	if(dep[x]<dep[y])swap(x,y);
	for(int i=20;i>=0;--i){
		if(dep[fa[x][i]]>=dep[y])
			x=fa[x][i];
	}
	if(x==y){return y;}
	for(int i=20;i>=0;--i){
		if(fa[x][i]!=fa[y][i]){
			x=fa[x][i];
			y=fa[y][i];
		}
	}
	return fa[x][0];
}
void dfs(int fro,int fat){
	dep[fro]=dep[fat]+1;
	fa[fro][0]=fat;
	for(int i=1;i<=20;++i)
		fa[fro][i]=fa[fa[fro][i-1]][i-1];
	for(int i=0;i<e[fro].size();++i){
		int to=e[fro][i];
		if(to==fa[fro][0])continue;
		dfs(to,fro);
	}
	return;
}
int main(){
	cin>>n>>m>>s;
	for(int i=1;i<n;++i){
		int x,y;cin>>x>>y;
		e[x].push_back(y);
		e[y].push_back(x);
	}
	dfs(s,0);
	for(int i=1;i<=m;++i){
		int a,b;cin>>a>>b;
		cout<<lca(a,b)<<endl;
	}
	return 0;
}
void dfs(int fro,int fat){
	dep[fro]=dep[fat]+1;
	fa[fro][0]=fat;
	for(int i=1;i<=20;++i)
		fa[fro][i]=fa[fa[fro][i-1]][i-1];
	for(int i=0;i<e[fro].size();++i){
		int to=e[fro][i];
		if(to==fa[fro][0])continue;
		dfs(to,fro);
	}
	return;
}
int main(){
	cin>>n>>m>>s;
	for(int i=1;i<n;++i){
		int x,y;cin>>x>>y;
		e[x].push_back(y);
		e[y].push_back(x);
	}
	dfs(s,0);
	for(int i=1;i<=m;++i){
		int a,b;cin>>a>>b;
		cout<<lca(a,b)<<endl;
	}
	return 0;
}
2020/7/22 14:54
加载中...