小字辈（左子右兄加强版）

苏慕TRYACE

于 2024-03-23 16:39:06 发布

阅读量483

点赞数 4

CC 4.0 BY-SA版权

文章标签：深度优先算法图论 C++ 二叉树 dfs

本文链接：https://blog.csdn.net/WJTF2/article/details/136969950

本文介绍了一个编程题目，涉及构建二叉树并使用深度优先搜索（DFS）算法找出每个节点的最小辈分。主要步骤包括构建家谱、遍历节点并更新辈分，最后输出结果。

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

文章目录

题目描述
思路
AC代码
总结

题目描述

在这里插入图片描述

思路

dfs + 二叉树

存储结构
1.用结构体数组存储每个节点的父节点、左右节点
2.用vector存储答案

具体做法
1.根据输出，构建家谱
2.遍历每一个点，找出父节点==0的点，该点为祖宗节点
3.从祖宗节点开始dfs
①当辈分大于上一次递归辈分时，说明找到了更小辈分的，将res清空，并将当前辈分最小的加入res
②当辈分等于上一次递归辈分时，将该节点加入res数组
4.如果该点的左节点存在（子节点），继续遍历，辈分+1
5.如果该点的右节点存在（兄弟节点），继续遍历，辈分不变

AC代码

#include <bits/stdc++.h>
using namespace std;
const int N = 100010;
vector<int> res;
typedef struct
{
    int fa, son, bro;
}person;
person p[N];
int cnt = -1;
void dfs(int x, int step)
{
    if(step > cnt)
    {
        cnt = step;
        res.clear();
        res.push_back(x);
    }
    else if(step == cnt) res.push_back(x);
    if(p[x].son) dfs(p[x].son, step + 1); //搜索子节点，辈分+1
    if(p[x].bro) dfs(p[x].bro, step); //搜索兄弟节点，辈分不变
}
int main()
{
    int n;
    scanf("%d", &n);
    for(int i = 0; i < n; i ++)
    {
        int a, b, c;
        scanf("%d%d%d", &a, &b, &c);
        if(b)
        {
            p[a].son = b;
            p[b].fa = a;
        }
        if(c)
        {
            p[a].bro = c;
            p[c].fa = a; //这里的意思不是c的父亲是a，只是在树的关系中，c是a的一个节点，这会影响到下面找祖宗节点
        }
    }
    int root;
    for(int i = 1; i <= n; i ++)
    {
        if(!p[i].fa)
        {
            root = i;
            break;
        }
    }
    //printf("%d\n", root);
    dfs(root, 1);
    printf("%d\n", cnt);
    sort(res.begin(), res.end());
    for(int i = 0; i < res.size(); i ++)
    {
        if(i != res.size() - 1) printf("%d ", res[i]);
        else printf("%d\n", res[i]);
    }
    return 0;
}