ICode9

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

LCA(树上倍增)

2022-07-31 20:33:43  阅读:82  来源: 互联网

标签:now 倍增 MAX 结点 long fa depth LCA 树上


https://www.luogu.com.cn/problem/P3379

  • 链式前向星存边
  • fa[i][j]  代表从i结点向上找 2^i 代的父亲,(i=0代表真父亲)
  • dfs从根结点开始fa[now][i] = fa[fa[now][i - 1]][i - 1];代表当前结点的第2^i代父节点是当前结点2^(i-1)父节点的2^(i-1)代父节点,然后再对其连接到的非父结点dfs
  • depth记录深度,用于判断循环的次数即log2(depth【x】)
#include <bits/stdc++.h>
using namespace std;
#define MAX 1000001
long long idx;
long long head[MAX];
long long depth[MAX];
long long fa[MAX][21];
long long n, m, s;
long long lg[MAX];
struct node
{
    long long to, nex;
} edges[MAX];
void add(long long u, long long v)
{
    edges[++idx].to = v;
    edges[idx].nex = head[u];
    head[u] = idx;
}

void input()
{
    cin >> n >> m >> s;

    for (long long i = 0; i < n - 1; i++)
    {
        long long a, b;
        scanf("%lld%lld", &a, &b);
        add(a, b);
        add(b, a);
    }
}
void dfs(long long now, long long faer)
{
    fa[now][0] = faer;
    depth[now] = depth[faer] + 1;
    for (long long i = 1; i <= log2(depth[now]); i++)
    {
        fa[now][i] = fa[fa[now][i - 1]][i - 1];
    }
    for (long long i = head[now]; i; i = edges[i].nex)
    {
        if (edges[i].to != faer)
        {
            dfs(edges[i].to, now);
        }
    }
}

long long LCA(long long a, long long b)
{
    if (depth[a] < depth[b])
    {
        swap(a, b);
    }
    while (depth[a] > depth[b])
    {
        a = fa[a][lg[depth[a] - depth[b]] - 1];
    }
    if (a == b)
    {
        return a;
    }
    for (long long i = lg[depth[a]] - 1; i >= 0; i--)
    {
        if (fa[a][i] != fa[b][i])
        {
            a = fa[a][i];
            b = fa[b][i];
        }
    }
    return fa[a][0];
}
void init_log2()
{
    for (long long i = 1; i <= n; ++i)
        lg[i] = lg[i - 1] + (1 << lg[i - 1] == i);
}
void main_work()
{
    for (long long i = 0; i < m; i++)
    {
        long long a, b;
        scanf("%lld%lld", &a, &b);
        printf("%lld\n", LCA(a, b));
    }
}
void print_log2()
{
    for (long long i = 0; i <= n; i++)
    {
        printf("%lld ", lg[i]);
    }
}
int main()
{
    input();
    init_log2();
    // print_log2();
    dfs(s, 0);
    main_work();
}

 

标签:now,倍增,MAX,结点,long,fa,depth,LCA,树上
来源: https://www.cnblogs.com/Wang-Xianyi/p/16538014.html

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

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

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

ICode9版权所有