【嘎】树-二叉树的直径

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 len = 0;
        if (root != null) {
            if (root.left != null) {
                len += getLength(root.left, 1);
            }
            if (root.right != null) {
                len += getLength(root.right, 1);
            }
        }
        return len;
    }

    public int getLength(TreeNode node, int len) {
        int leftlen = 0;
        int rightlen = 0;

        if (node.left != null) {
            leftlen = getLength(node.left, len);
        }
        if (node.right != null) {
            rightlen = getLength(node.right, len);
        }
        if (leftlen > rightlen) {
            len += leftlen;
        } else {
            len += rightlen;
        }
        return len;
    }

}

然后想用递归，但是return的时候只能return我想要的左子树或者右子树中的大者，还有当前最大的左子树加右子树。。陷入了。。然后评论中有人用了全局变量来存放max

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    int max = 0;
    public int diameterOfBinaryTree(TreeNode root) {
        if (root != null) {
            // 遍历每一个节点，求出此节点作为根的深度，那么左子树深度加右子树深度的最大值即是答案
            setDepth(root);
            return max;
        }
        return 0;
    }

    public int setDepth(TreeNode root) {
        if (root != null) {
            int left = setDepth(root.left);
            int right = setDepth(root.right);
            if (right + left > max) 
                max = right + left;
            return Math.max(right, left) + 1;
        }
        return 0;
    }

}

1. 二叉树的直径不一定过根节点，因此需要去搜一遍以每个节点为root的子树对应的直径，取最大值。
2. root的直径 = root左子树高度 + root右子树高度
3. root的高度 = max {root左子树高度, root右子树高度} + 1