2017 小米笔试题 编程题 求树的高度 Java代码实现

2017 小米笔试题 编程题 求树的高度 Java代码实现

题目:树的高度

时间限制:C/C++语言 1000MS;其他语言 3000MS
内存限制:C/C++语言 65536KB;其他语言 589824KB
题目描述:
现在有一棵合法的二叉树,树的节点都是用数字表示,
现在给定这棵树上所有的父子关系,求这棵树的高度

输入
输入的第一行表示节点的个数n(1<=n<=1000,节点的编号为0到n-1)组成,
下面是n-1行,每行有两个整数,第一个数表示父节点的编号,第二个数表示子节点的编号

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

样例输入
5
0 1
0 2
1 3
1 4
样例输出

3

Java简单实现的代码

import java.util.HashMap;
import java.util.Map;
import java.util.Scanner;

public class Main {

    public static void main(String[] args) {
        // TODO Auto-generated method stub
        Scanner sc = new Scanner(System.in);
        int nodeNum = sc.nextInt();
        Map<Integer, Integer> dict = new HashMap<>();
        int father, son ;
        int highest = 0;

        if(nodeNum == 1) {
            System.out.print(1);
            return;
        }
        for (int i=0; i< nodeNum -1; i++){
            father = sc.nextInt();
            son = sc.nextInt();
            dict.put(son, father); // 反序存入
        }
        int count;
        for(Integer key : dict.keySet() ){
            count = 1;
            //  循环获得每个节点的高度
            while(dict.containsKey(key)){
                count ++;
                key = dict.get(key);
            }
            if(highest < count) highest = count;
        }
        System.out.print(highest);
    }
}
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值