543.diameter-of-binary-tree

给定一棵二叉树，你需要计算它的直径长度。一棵二叉树的直径长度是任意两个结点路径长度中的最大值。这条路径可能穿过也可能不穿过根结点。

示例 :
给定二叉树

返回 3, 它的长度是路径 [4,2,1,3] 或者 [5,2,1,3]。

注意 ：两结点之间的路径长度是以它们之间边的数目表示。

解法

首先要计算每个节点的路径长度,路径长度 = 左节点的深度+右节点的深度.

由于可能存在重复计算,因此要使用备忘录.

最后,在递归的时候,更新结果就行

class Solution {
        private int max = 0;
		
		private Map<TreeNode, Integer> r = new HashMap<>();
		
		public int diameterOfBinaryTree(TreeNode root) {

			if (null == root) {
				return 0;
			}

			
			int l = depth(root.left);
			int r = depth(root.right);
			max = Math.max(max, l + r);
			
			diameterOfBinaryTree(root.left);
			diameterOfBinaryTree(root.right);

			return max ;

		}

		private int depth(TreeNode root) {
			
			if(r.containsKey(root)) {
				return r.get(root);
			}
			
			if (null == root) {
				return 0;
			}
			
			r.put(root, Math.max(depth(root.left), depth(root.right)) + 1);
			return r.get(root);
		}
}