ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

倍增求LCA

2021-03-01 14:05:11  阅读:177  来源: 互联网

标签:now int fathers tot depth edges LCA 倍增


参考:倍增求LCA

#include <iostream>
#include <cstdlib>
#include <vector>
#include <queue>

using namespace std;

#define SIZE 500005

struct edge{
    int to;
    int next;
}edges[SIZE<<1];
int head[SIZE];
int depth[SIZE];
int fathers[SIZE][21];
int lg[SIZE];
int tot=0;

void add(int from, int to){
    tot++;
    edges[tot].to = to;
    edges[tot].next = head[from];
    head[from] = tot;
}

void dfs(int now, int father){
    fathers[now][0] = father;
    depth[now] = depth[father] + 1;
    for(int i=1; i<lg[depth[now]]; i++){
        fathers[now][i] = fathers[fathers[now][i-1]][i-1];
    }
    for(int i=head[now]; i; i=edges[i].next){
        if(edges[i].to!=father)
            dfs(edges[i].to, now);
    }
}

void swap(int&u, int& v){
    int temp = u;
    u = v;
    v = temp;
}

int LCA(int u, int v){
    if(depth[u]<depth[v])   swap(u, v);
    while(depth[u]>depth[v]){
        u = fathers[u][lg[depth[u]-depth[v]]-1];
    }
    if(u==v)    return u;
    for(int i=lg[depth[u]]-1; i>=0; i--){
        if(fathers[u][i]!=fathers[v][i]){
            u = fathers[u][i];
            v = fathers[v][i];
        }
    }
    return fathers[u][0];
}

int main() {
    int M,N,S,u,v;
    cin >> M >> N >> S;
    for(int i=1; i<M; i++){
        cin >> u >> v;
        add(u,v);
        add(v,u);
    }
    for(int i=1; i<=N; i++){预先算出log_2(i)+1的值,用的时候直接调用就可以了
        lg[i] = lg[i-1] + (1<<lg[i-1]==i);
    }
    dfs(S,0);
    int a = 234;
    for(int i=1; i<=N; i++){
        cin >> u >> v;
//        cout << u << " " << v << " " << LCA(u,v) << endl;
        cout << LCA(u,v) << endl;
    }
    return 0;
}
/*
5 5 1
3 1
2 4
5 1
1 4
2 4
 *
 * */

标签:now,int,fathers,tot,depth,edges,LCA,倍增
来源: https://blog.csdn.net/weixin_43111232/article/details/114261378

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有