题目描述
现在有一棵合法的二叉树,树的节点都是用数字表示,现在给定这棵树上所有的父子关系,求这棵树的高度
输入描述:
输入的第一行表示节点的个数n(1 ≤ n ≤ 1000,节点的编号为0到n-1)组成, 下面是n-1行,每行有两个整数,第一个数表示父节点的编号,第二个数表示子节点的编号
输出描述:
输出树的高度,为一个整数
示例1
输入
5 0 1 0 2 1 3 1 4
输出
3
这道题开始我是想到用tree型结构来做,但是代码写到后来编不下去了,感觉这条路行不通(因为题目中需要查找父节点)。于是看了其他人的思路,发现有人用HashMap来做,构造两个hashmap分别来存放父节点和其子节点的个数,子节点以及子节点的深度。其实就是用了类似于数组索引的思想。//做题时,不要被其表面所迷惑,要先自己想想还有没有可能用更方便的方法
public static void main(String[] args){Scanner sc = new Scanner(System.in);
while(sc.hasNext()){
int num = sc.nextInt();
HashMap<Integer,Integer> tree = new HashMap();
HashMap<Integer,Integer> child_deep = new HashMap();
//树一开始根结点为0,孩子结点个数为0
tree.put(0,0);
//孩子结点为0,深度为1
child_deep.put(0,1);
int dep = 0;
int max = 1;
for (int i = 0;i<num-1;i++){
int father = sc.nextInt();
int son = sc.nextInt();
//ArrayList里面是contains
//树里面不包含父节点或者儿子已满的情况,忽略
if(!tree.containsKey(father) || tree.get(father) >=2 ){
continue;
}
//父亲的孩子数+1
tree.put(father,tree.get(father)+1);
//孩子变为另一个父亲
tree.put(son,0);
//深度+1
child_deep.put(son,child_deep.get(father)+1);
dep = child_deep.get(son);
if(max<dep ){
max = dep;
}
}
System.out.println(max);
}
}