最近公共祖先

百度2021笔试 c++卷(LCA)

这个题是一道非常非常经典的LCA问题，我解释下思路。

首先假如我们是求树上两个最远点的距离那么很简单，直接递归找最大深度就可以了，但是本题要求我们找任意两点的最短路，这下我们必须用到LCA来求解。首先要是直接找红色到绿色的距离是很难计算的， 这个题我们转化为求红色点到根的距离 + 绿色点到根的距离 - 两倍的粉色距离。 经过这样的一步转化，我们在lca的时候只要保存每个点到根的距离，就可以在o(1)时间解决问题。

下面是一个标准的lca树上倍增代码， 至于向上标记法我就不写了，太过简单，同学们可以自行百度。

#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
const int N = 10010, M = N * 2 + 10;

int h[N], e[M], w[M], ne[M], idx;

int f[N][16], q[N], depth[N]; //f[j][k] 表示的是j节点向上跳跃2^k步到达的节点
int n, m ;
int dist[N]; //保存每一个节点到根距离

void add(int a, int b, int c)  // 添加一条边a->b，边权为c
{
    e[idx] = b, w[idx] = c, ne[idx] = h[a], h[a] = idx ++ ;
}

void bfs()
{
    memset(depth, 0x3f, sizeof depth);
    int hh = 0, tt = 1;
    q[++ tt] = 1;
    depth[0] = 0, depth[1] = 1;
    while (hh <= tt) 
    {
        auto t = q[hh ++];
        for (int i = h[t]; ~i; i = ne[i])
        {
            int j = e[i];
            if (depth[j] > depth[t] + 1) {
                depth[j] = depth[t] + 1;
                dist[j] = dist[t] + w[i];
                q[++ tt ] = j;
                f[j][0] = t;
                for (int k = 1; k <= 14; k ++)
                {
                    f[j][k] = f[f[j][k - 1]][k - 1];
                }
            }
        }
    }
    
   
}


int lca(int a, int b)
{
    if (depth[a] < depth[b]) swap(a, b);
    for (int i = 14; i >= 0; i --) {
        if (depth[f[a][i]] >= depth[b]) {
            a = f[a][i];
        }
    }
    if (a == b) return a;
    for (int k = 14; k >= 0; k --)
    {
        if (f[a][k] != f[b][k]) { //这里是跳到了LCA的子节点，也就是相遇前的最后一个点
            a = f[a][k], b = f[b][k];
        }
    }
    return f[a][0];
}
int main()
{
    cin >> n >> m;
    memset(h, -1, sizeof h);
    for (int i = 0; i < n - 1; i ++ ) {
        int a, b, c;
        cin >> a >> b >> c;
        add(a, b, c);
        add(b, a, c);
    }
    bfs();
    
    while (m -- ) 
    {
        int a, b;
        cin >> a >> b;
        
        int p = lca(a, b);
        cout << dist[a] + dist[b] - 2 * dist[p] << endl;
    }
    
    return 0;
    
}

本题是一道较好的lca题目，也是我们笔试中遇到的困难级别了，希望大家好好掌握。