社区讨论

Unac100求助

P3379【模板】最近公共祖先(LCA)参与者 4已保存回复 8

讨论操作

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

当前回复
8 条
当前快照
1 份
快照标识符
@m0do2sy6
此快照首次捕获于
2024/08/28 17:41
2 年前
此快照最后确认于
2025/11/04 22:10
4 个月前
查看原帖
开O2时TLE#11,#12,#13,关O2时TLE#14,想知道有没有没有优化方案。
代码偏模板,可能有雷同
CPP
#include<bits/stdc++.h>
#define int long long
using namespace std;
const int N=5e5+110;
int d[N],jump[N][21];
vector<int> v[N];
void dfs(int x,int fa,int sum){
	jump[x][0]=fa;
	d[x]=sum;
	for(int i=0;i<v[x].size();i++) if(v[x][i]!=fa) dfs(v[x][i],x,sum+1);
}
int lca(int x,int y){
	if(d[x]<d[y]) swap(x,y);
	for(int i=20;i>=0;i--) if(d[jump[x][i]]>=d[y]) x=jump[x][i];
	if(x==y) return x;
	for(int i=20;i>=0;i--) if(jump[x][i]!=jump[y][i]) x=jump[x][i],y=jump[y][i];
	return jump[x][0];
}
signed main(){
	int n,t,s;
	cin>>n>>t>>s;
	for(int i=1;i<n;i++){
		int x,y;
		cin>>x>>y;
		v[y].push_back(x);
		v[x].push_back(y);
	}
	dfs(s,0,1);
	for(int i=1;i<=20;i++) for(int j=1;j<=n;j++) jump[j][i]=jump[jump[j][i-1]][i-1];
	while(t--){
	    int a,b;
	    cin>>a>>b;
	    cout<<lca(a,b)<<endl;
	}
	return 0;
}

回复

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

正在加载回复...