社区讨论

倍增lca样例过了0分求调

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

讨论操作

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

当前回复
3 条
当前快照
1 份
快照标识符
@m2uf2we1
此快照首次捕获于
2024/10/29 20:21
去年
此快照最后确认于
2025/11/04 15:44
4 个月前
查看原帖
CPP
#include <bits/stdc++.h>
using namespace std;

int tree[500005][30];//i点2^j祖先
int  deep[500005];
int n,m,s;
void bd(int cur){
    if (cur==s||deep[cur]) return;
    bd(tree[cur][1]);
    deep[cur]=deep[tree[cur][1]]+1;
}
void build(){
	deep[0]=-114514;
    int a,b;
    for (int i=1;i<n;i++){
        cin>>a>>b;
        tree[a][1]=b;
    }
    for (int i=2;i<=25;i++){
        for (int j=1;j<=n;j++){
            tree[j][i]=tree[tree[j][i-1]][i-1];
        }
    }
    for (int i=1;i<=n;i++) bd(i);
    return;
}
int todeep(int num,int dp){
    if (deep[num]==dp) return num;
    int i=25;
    while (deep[tree[num][i]]<dp) i--;
    return todeep(tree[num][i],dp);
}
int c(int a,int b){
	if (a!=b&&tree[a][1]==tree[b][1]) return tree[b][1];
    int i=25;
    while (tree[a][i]==tree[b][i]&&i>=1) i--;
    return c(tree[a][i],tree[b][i]);
}
int lca(int s,int b){
    b=todeep(b,deep[s]);
    if (s==b) return s;
    return c(s,b);
}
signed main(){
	ios::sync_with_stdio(0);
	cin.tie(0);
    int a,b;
    cin>>n>>m>>s;
    build();
    while (m--){
        cin>>a>>b;
        if (deep[a]<deep[b]) cout<<lca(a,b)<<"\n";
        else cout<<lca(b,a)<<"\n";
    }
    return 0;
}
先跳到一级再一起往上跳,倍增

回复

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

正在加载回复...