LeetCode Java刷题笔记—543. 二叉树的直径

Posted 2022-04-19 刘Java

543. 二叉树的直径

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

简单难度。这道题和124. 二叉树中的最大路径和（困难）思路是一样的，但却是简单难度。主要注意的是这条路径可能穿过也可能不穿过根结点，我们维护一个全局变量max，然后遍历每一个节点，求该节点左子树和右子树的高度和，将该题转换为求二叉树的每个节点的左右子树的高度和的最大值。

private int max = 0;

public int diameterOfBinaryTree(TreeNode root) 
    depth(root);
    return max;


private int depth(TreeNode root) 
    if (root == null) 
        return 0;
    
    //左右子树的高度
    int left = depth(root.left);
    int right = depth(root.right);
    //此前的节点的最大直径和当前节点的直径比较，取最大值
    max = Math.max(max, left + right);
    //返回左右子树的高度+1
    return Math.max(left, right) + 1;