【编程题m_0032】树的高度（小米2017秋招真题）

2019独角兽企业重金招聘Python工程师标准>>>

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

输入 输入的第一行表示节点的个数n（1<=n<=1000，节点的编号为0到n-1）组成， 下面是n-1行，每行有两个整数，第一个数表示父节点的编号，第二个数表示子节点的编号	样例输入 5 0 1 0 2 1 3 1 4
输出 输出树的高度，为一个整数	样例输出 3

解题思路：采用数组存储各节点，编号对应数组下标，数组元素的值为父节点编号；
 如果数组长度为1，表示只有一个节点，故树的高度为1；否则，从后向前依次遍历，以该节点为叶子节点时，数的高度，比较取最大值即为整棵树的高度。

package BiShiTi;

import java.util.Scanner;

/**
 * 解题思路：采用数组存储各节点，编号对应数组下标，数组元素的值为父节点编号；
 * 如果数组长度为1，表示只有一个节点，故树的高度为1；否则，从后向前依次遍历，以该节点为叶子节点时，数的高度，比较取最大值即为整棵树的高度。
 */
public class m_0032 {
	public static void main(String [] args){
		Scanner sc = new Scanner(System.in);
		int n = sc.nextInt();
		
		int [] tree = new int[n];
		tree[0] = -1;
		for(int i = 1; i < n; i++){
			tree[i] = -1;
		}
		
		for(int i = 1; i < n; i ++) {
			int a = sc.nextInt();
			int b = sc.nextInt();
			tree[b] = a;
		}
		
		System.out.println(getTreeHight(tree));
	}
	
	public static int getTreeHight(int [] _tree){
		int [] tree = _tree;
		int hight = 0;
      	if(tree.length == 1){
        	return 1;
        }
		for(int i = tree.length-1; i > 0; i--){
			int tempIndex = i;
			int tempHight = 0;
			while (tempIndex != -1) {
				tempHight ++;
				tempIndex = tree[tempIndex];
			}
			hight = hight > tempHight?hight:tempHight;
		}
		return hight;
	}
}

 

转载于:https://my.oschina.net/MasterLi161307040026/blog/1842424