1172. 祖孙询问

最新推荐文章于 2023-01-14 16:28:03 发布

追寻远方的人

最新推荐文章于 2023-01-14 16:28:03 发布

阅读量144

点赞数

分类专栏：算法提高复习文章标签：图论算法 LCA

本文链接：https://blog.csdn.net/segegse/article/details/125353251

版权

算法提高复习专栏收录该内容

167 篇文章 3 订阅

订阅专栏

该博客介绍了一种处理有根无向树中节点祖孙关系查询的方法。通过使用LCA（最近公共祖先）算法，可以高效地回答给定节点对之间的关系，输出1表示查询节点是祖先，2表示是子孙，0表示没有直接关系。代码实现中包含了树的构建、BFS遍历以计算节点的深度和最近公共祖先，并展示了具体的输入输出样例。

摘要由CSDN通过智能技术生成

给定一棵包含 n 个节点的有根无向树，节点编号互不相同，但不一定是 1∼n。

有 m 个询问，每个询问给出了一对节点的编号 x 和 y，询问 x 与 y 的祖孙关系。

输入格式

输入第一行包括一个整数表示节点个数；

接下来 n 行每行一对整数 a 和 b，表示 a 和 b 之间有一条无向边。如果 b 是 −1，那么 a 就是树的根；

第 n+2 行是一个整数 m 表示询问个数；

接下来 m 行，每行两个不同的正整数 x 和 y，表示一个询问。

输出格式

对于每一个询问，若 x 是 y 的祖先则输出 1，若 y 是 x 的祖先则输出 2，否则输出 0。

数据范围

1≤n,m≤4×104,
1≤每个节点的编号≤4×104

输入样例：

输出样例：

思路：

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-nmtJ9md7-1655566893834)(1172.%20%E7%A5%96%E5%AD%99%E8%AF%A2%E9%97%AE.assets/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzQxMjgwNjAw,size_16,color_FFFFFF,t_70-16526899167952.png)]

代码：

#include <bits/stdc++.h>
using namespace std;
const int N = 4e4 + 10, M = 2 * N;
int h[N], ne[M], e[M], idx;
int fa[N][16], depth[N];
int root, n, m;

void add(int a, int b)
{
    e[idx] = b;
    ne[idx] = h[a];
    h[a] = idx++;
}

void bfs(int root)
{
    // 哨兵depth[0] = 0: 如果从i开始跳2^j步会跳过根节点
    // fa[fa[j][k-1]][k-1] = 0
    // 那么fa[i][j] = 0 depth[fa[i][j]] = depth[0] = 0
    depth[0] = 0;
    depth[root] = 1;
    queue<int> qu;

    qu.push(root);
    while (qu.size())
    {
        int t = qu.front();
        qu.pop();

        for (int i = h[t]; i != -1; i = ne[i])
        {
            int j = e[i];
            if (!depth[j])
            {
                depth[j] = depth[t] + 1;
                qu.push(j);

                fa[j][0] = t;
                for (int k = 1; k <= 15; k++) // 从小到大 递推
                    fa[j][k] = fa[fa[j][k - 1]][k - 1];
            }
        }
    }
}

int lca(int a, int b)
{
    if (depth[a] < depth[b])
        swap(a, b);
    for (int k = 15; k >= 0; k--) // 从大到小
        if (depth[fa[a][k]] >= depth[b])
            a = fa[a][k];

    if (a == b)
        return a;

    for (int k = 15; k >= 0; k--)
    {
        if (fa[a][k] != fa[b][k])
        {
            a = fa[a][k];
            b = fa[b][k];
        }
    }
    return fa[a][0];
}

int main()
{
    scanf("%d", &n);

    memset(h, -1, sizeof(h));
    for (int i = 0; i < n; i++)
    {
        int a, b;
        scanf("%d %d", &a, &b);
        if (b == -1)
            root = a;
        else
            add(a, b), add(b, a);
    }

    bfs(root);

    scanf("%d", &m);
    for (int i = 0; i < m; i++)
    {
        int a, b;
        scanf("%d %d", &a, &b);
        int p = lca(a, b);
        if (p == a)
            puts("1");
        else if (p == b)
            puts("2");
        else
            puts("0");
    }

    return 0;
}