小米笔试题：树的高度（非给一棵树结构，而是根据输入计算）

最新推荐文章于 2024-09-13 23:06:06 发布

NO1._LUA

最新推荐文章于 2024-09-13 23:06:06 发布

阅读量268

点赞数 1

文章标签：二叉树

本文链接：https://blog.youkuaiyun.com/weixin_41613698/article/details/105654401

版权

这道小米笔试题要求根据输入的父子关系计算二叉树的高度。输入描述包含节点数量和父子关系，输出即为树的高度。解题关键是每次更新节点高度，避免构建实际树结构，时间复杂度为O(n^2)。

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

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

输入描述:

输入的第一行表示节点的个数n（1 ≤ n ≤ 1000，节点的编号为0到n-1）组成，
下面是n-1行，每行有两个整数，第一个数表示父节点的编号，第二个数表示子节点的编号

输出描述:

输出树的高度，为一个整数

示例1

输入
5
0 1
0 2
1 3
1 4

输出
3

分析

不用根据输入构造一棵二叉树，只需要记录下每一个结点的高度，子节点是父结点高度+1.
考虑到输入顺序可能不会从根节点开始，所以在每一行输入后都要对之前的结点高度进行更新
如果这一题输入的不是结点编号并且编号是从0~n，那么久不能用数组去记录高度了。可用hashmap。

时间复杂度
O（n^2）

代码

import java.util.Scanner;
public class Main{
    public static void main(String [] args){
        Scanner in = new Scanner(System.in);
        int n = in.nextInt();
        if(n < 3)System.out.print(n);
        int [] times = new int[n];
        int [] height = new int[n];
        int [] parents = new int[n-1];
        int [] children = new int[n-1];
        int max = 0;
        height[0] = 1;
        for(int i = 0;i < n-1;i++){
            parents[i] = in.nextInt();
            children[i] = in.nextInt();
            times[parents[i]] ++;
            if(times[parents[i]]<3){
                height[children[i]] = height[parents[i]]+1;
                //更新这一行前面的每一行的结点的高度
                for(int j=0;j<i;j++){
                    //依旧要排除多叉的情况
                    if(times[parents[j]]<3)
                        height[children[j]] = height[parents[j]]+1;
                }
            }
            //记录当前最深的孩子结点的高度
            max = Math.max(max, height[children[i]]);
        }
        System.out.print(max);
    }
}