LeetCodeQ543 二叉树直径 java 100%

思路

本题要求二叉树的直径，即任意两节点之间的最长路径。一般二叉树问题都需要进行问题的分解，将大问题分解为针对单个节点的子问题。所以，看到针对节点之间的路径问题，一般想到的都是两个叶子结点之间的最大路径，而叶子结点之间的最长路径肯定经过他们最高的祖先节点，且最长路径即两个叶节点到这个祖先节点的路径之和。因此问题转化子问题，求左右两个子树的深度，假设当前节点就是祖先节点，那么以它为根的树的直径就是左子树深度加右子树深度。这样，通过深度遍历每个节点，不断向上返回该子树的最大深度，针对每个节点都可以判断出它的直径，然后去最大直径即可。

代码

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    int max;
    public int diameterOfBinaryTree(TreeNode root) {
        max=0;
        if(root==null) return 0;
        sub(root);
        return max-1;
    }
    int sub(TreeNode root){
        if(root==null) return 0;
        int hl=sub(root.left);
        int hr=sub(root.right);
        max=max>(hl+hr+1)?max:(hl+hr+1);
        return 1+(hr>hl?hr:hl);
    }
}