POJ 1330 Nearest Common Ancestors

最新推荐文章于 2020-11-23 16:04:39 发布

yurio7r

最新推荐文章于 2020-11-23 16:04:39 发布

阅读量365

点赞数

CC 4.0 BY-SA版权

文章标签： LCA

本文链接：https://blog.youkuaiyun.com/MrSiz/article/details/49936721

图论_习题专栏收录该内容

3 篇文章

订阅专栏

本文介绍了一个基于深度优先搜索（DFS）实现的最低公共祖先（LCA）算法。该算法通过预处理的方式，记录每个节点的父节点及深度，进而快速查询任意两个节点的最低公共祖先。适用于解决树形结构中的LCA问题。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

模板题。。。。

#include <iostream>
#include <vector>
#include <cstdio>

using namespace std;

const int MAX=10005;
vector<int> G[MAX];
int depth[MAX];
int parent[MAX];
int root, n;
bool vis[MAX];

void dfs(int v, int p, int d)
{
  parent[v] = p;
  depth[v] = d;
  for (int i=0;i<G[v].size(); i++)
  {
    if (G[v][i] != p)
      dfs(G[v][i], v, d+1);
  }
}

void init()
{
  for(int i = 1; i <= n; i++)
    if(!vis[i])
      dfs(i, -1, 0);
}

int lca(int u, int v)
{
  while (depth[u] > depth[v])
    u = parent[u];
  while (depth[v] > depth[u])
    v = parent[v];
  while (u != v)
  {
    u = parent[u];
    v = parent[v];
  }
  return u;
}

int main()
{
  std::ios::sync_with_stdio(false);
  int T;
  cin >> T;
  while (T--)
  {
    cin >>  n;
    for (int i = 1; i <= n; i++)
      G[i].clear();
    for (int i = 0; i < n-1; i++)
    {
      int t1,t2;
      cin >> t1 >> t2;
      G[t1].push_back(t2);
      vis[t2] = true;
    }
    int f_u, f_v;
    cin >> f_u >> f_v;
    //root = 1;
    //dfs(1, -1, 0);
    init();
    //cout << depth[f_v] << depth[f_u] << endl;
    cout << lca(f_u,f_v) << endl;
  }
  return 0;
}