Solution 11: 二叉树中节点的最大距离

问题描述

RT. 如果把二叉树看做是一张图，父子节点的连线看成是双向的，距离的定义为两个节点之间变得数目。

解决思路

求根节点的左子树的高度和右子树的高度，相加。

程序

public class MaxDistanceInTree {
	public int getMaxDistanceInTree(TreeNode root) {
		if (root == null) {
			return 0;
		}

		int leftHeight = getHeightOfTree(root.left);
		int rightHeight = getHeightOfTree(root.right);

		return leftHeight + rightHeight;
	}

	private int getHeightOfTree(TreeNode root) {
		if (root == null) {
			return 0;
		}

		return Math
				.max(getHeightOfTree(root.left), getHeightOfTree(root.right)) + 1;
	}
}