由整数对(父节点和子节点的关系)组成的二叉树的高度问题

问题描述:

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

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

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

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

样例输出: 3

 

解答思路:

创建一个map<节点,所在的层>,那么map的values中的最大值即为二叉树的高度。

import java.util.*;
public class Main {
public static void main(String[] args) {
Scanner in = new Scanner(System.in);
while (in.hasNextInt()) {
Map<Integer,Integer> map = new HashMap<>();
int n = in.nextInt();
for (int i = 0; i<n-1; i++){
int a = in.nextInt();
int b = in.nextInt();
if (map.containsKey(a)){
map.put(b,map.get(a)+1);
}else {
map.put(a,1);
map.put(b,2);
}
}
Collection<Integer> values = map.values();
int max = 0;
for (int value : values){
if (value>0){
max = value;
}
}
System.out.println(max);
}
}
}


ps:不知这种思路是否正确呢?笔试完了才想出来...
posted @ 2016-09-23 21:21  小样儿1020  阅读(1820)  评论(0编辑  收藏  举报