代码题--C++--牛客网小米求二叉树的高度

最新推荐文章于 2024-09-13 17:03:04 发布

原创最新推荐文章于 2024-09-13 17:03:04 发布 · 237 阅读

1 ·

本内容遵循CC 4.0 BY-SA版权协议

收录于

C++

剑指offer

面试题

此博客围绕牛客网小米的代码题展开，题目是给定一棵合法树的所有父子关系，用数字表示节点，要求计算树的高度。解题思路是用数组记录有效节点高度和孩子数量，先设节点最低高度为1，根据输入数据更新高度，比较保留最大值并输出。

代码题--C++--牛客网小米求二叉树的高度

题目描述

现在有一棵合法的树，树的节点都是用数字表示，现在给定这棵树上所有的父子关系，求这棵树的高度。

输入描述:
输入的第一行表示节点的个数n（1 ≤ n ≤ 1000，节点的编号为0到n-1）组成，
下面是n-1行，每行有两个整数，第一个数表示父节点的编号，第二个数表示子节点的编号
输出描述:
输出树的高度，为一个整数
示例1
输入
5
0 1
0 2
1 3
1 4
输出
3

解题思路：

用一个数组nodes记录有效节点的高度，用数组childnum记录每个位置节点的孩子数量，先保存nodes的高度最低为1，然后数如的没一对数据，第一个都代表父节点，所以记录对应高度为nodes[f]+1，将次高度h保存在子节点的有效高度，然后比较最大的，保留并输出。

代码如下：

#include <iostream>
using namespace std;
#include <vector>

int main()
{
    int n,H = 1;
    int i = 0;
    int f,c, h;
    vector<int> nodes(1000, 0);    //有效节点的高度
    nodes[0] = 1; // 题目说了至少有一个节点，高度只是是1
    vector<int> childnum(1000, 0); //记录节点的孩子数量
    cin >> n;
    while(--n){
        cin >> f >> c;
        //父节点不存在 或者 父节点已有两个子节点 就跳过
        if(nodes[f]==0 || childnum[f] == 2)
            continue;
        childnum[f] += 1;
        h = nodes[f] + 1;
        nodes[c] = h;
        if(h > H) H = h;
    }
    cout << H;
    return 0;
}