二叉树的直径

题目来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/diameter-of-binary-tree

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

示例 :

给定二叉树

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

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

由于本题中的树节点数据没有用，所以可以修改原本的值，表示为该节点最大的直径。代码如下：

 /**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public int diameterOfBinaryTree(TreeNode root) {
        int[] ans = {0};
        setLen(root,ans);
        return ans[0];
    }

    private void setLen(TreeNode root,int[] ans){
        if(root == null)
            return;
        setLen(root.left,ans);
        setLen(root.right,ans);

        TreeNode left = root.left;
        TreeNode right = root.right;

        if(left == null && right == null){
            root.val = 0;
        }else if(left != null && right != null){
            root.val = Math.max(left.val,right.val) + 1;
            ans[0] = Math.max(ans[0],2+left.val+right.val);
        }else if(left == null){
            root.val = right.val + 1;
        }else{
            root.val = left.val + 1;
        }
        ans[0] = Math.max(ans[0],root.val);   
    }
}