社区讨论
lca tle*3 敢问dalao如何优化
P3379【模板】最近公共祖先(LCA)参与者 13已保存回复 32
讨论操作
快速查看讨论及其快照的属性,并进行相关操作。
- 当前回复
- 32 条
- 当前快照
- 1 份
- 快照标识符
- @mi6m6p30
- 此快照首次捕获于
- 2025/11/20 07:09 4 个月前
- 此快照最后确认于
- 2025/11/20 07:30 4 个月前
CPP
#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
const int maxn=500001;
int n,m,a,b,head[maxn],deep[maxn],p[maxn][31/*大概*/],num=0,vis[maxn];
struct EDGE{
int next,to;
}edge[2*maxn];
int read()
{
int x=0;char ch=getchar();
while(ch<'0'||ch>'9')ch=getchar();
while(ch>='0'&&ch<='9')x=x*10+ch-'0',ch=getchar();
return x;
}
void add(int from,int to){
edge[++num].next=head[from];
edge[num].to=to;
head[from]=num;
}
void dfs(int x){
for(int i=head[x];i;i=edge[i].next){
if(vis[edge[i].to]!=1){
vis[edge[i].to]=1;
deep[edge[i].to]=deep[x]+1;
p[edge[i].to][0]=x;
dfs(edge[i].to);
}
}
}
int lca(int a,int b){
int i,j;
if(deep[a]<deep[b]){
int t=a;a=b;b=t;
}
for(i=0;(1<<i)<=deep[a];i++);
i--;
for(j=i;j>=0;j--){
if(deep[a]-(1<<j)>=deep[b]){
a=p[a][j];
}
}
if(a==b)return a;
for(j=i;j>=0;j--){
if(p[a][j]!=p[b][j]&&p[a][j]!=-1/*也许不需要*/){
a=p[a][j];b=p[b][j];
}
}
return p[a][0];
}
int main(){
memset(deep,0,sizeof(deep));memset(p,-1,sizeof(p));memset(vis,0,sizeof(vis));
int d;n=read();m=read();d=read();
for(int i=1;i<=n-1;i++){
int u=read(),v=read();
add(u,v);add(v,u);
}
deep[d]=0;vis[d]=1;
dfs(d);
for(int j=1;j<=21;j++)
for(int i=1;i<=n;i++){
if(p[i][j-1]!=-1){
p[i][j]=p[p[i][j-1]][j-1];
}
}
while(m--)a=read(),b=read(),cout<<lca(a,b)<<endl;
return 0;
}
回复
共 32 条回复,欢迎继续交流。
正在加载回复...