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.Scanner;

public class Solution {
	
	public static void main(String[] args) {
		Scanner scanner = new Scanner(System.in);
		while(scanner.hasNext()){
			int m = scanner.nextInt();
			TreeNode rootNode = new TreeNode(scanner.nextInt());
			createTree(rootNode, rootNode.val, scanner.nextInt());	
			for(int i=1;i<m-1;i++){
				createTree(rootNode, scanner.nextInt(), scanner.nextInt());
			}
			
			int height = getTreeHeight(rootNode);
			System.out.println(height);
		}
		scanner.close();
	}
	
	/**
	 * 求树的高度
	 * @param rootNode
	 * @return
	 */
	private static int getTreeHeight(TreeNode rootNode) {
		if(rootNode==null){
			return 0;
		}
		int nLeft = getTreeHeight(rootNode.left);
		int nRight = getTreeHeight(rootNode.right);
		return (nLeft>nRight)?(nLeft+1):(nRight+1);
	}

	/**
	 * 建树
	 * @param root
	 * @param father
	 * @param childVal
	 */
	public static void createTree(TreeNode root,int father,int childVal){
		if(root==null){
			return;
		}
		if(root!=null && root.val==father){
			if(root.left==null){//左孩子为空即将其作为左孩子节点
				root.left = new TreeNode(childVal);
			}else{//作为右孩子节点
				root.right = new TreeNode(childVal);
			}
			return;
		}
		
		createTree(root.left, father, childVal);
		createTree(root.right, father, childVal);
	}
}

/**
 * 对象： 树
 *
 */
class TreeNode{
	TreeNode left;
	TreeNode right;
	int val;
	
	public TreeNode(int val) {
		this.val = val;
	}
	
}