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 = １;            //  循环获得每个节点的高度            while(dict.containsKey(key)){                count ++;                key = dict.get(key);            }            if(highest < count) highest = count;        }        System.out.print(highest);    }}