【leetcode】543. 二叉树的直径

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

示例 :
给定二叉树

      1
     / \
    2   3
   / \     
  4   5    

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

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

解题思路基本跟官方思路一致。【官方解题思路】

• 计算最长路径
• 若横跨根节点，则应该是左右两边最大深度之和；
• 若不横跨根节点，则转化为求左子树的最长路径或者右子树的最长路径；
• 倒过来思考，则最长路径为左右子树最大深度之和与左子树、右子树最大路径里面的最大值。

我的代码：

class Solution(object):
    def max_depth(self, root):
        if not root:
            return 0

        R = self.max_depth(root.right)
        L = self.max_depth(root.left)
        self.ans = max(self.ans, L+R+1)
    
        return max(R, L) + 1

    def diameterOfBinaryTree(self, root):
        """
        :type root: TreeNode
        :rtype: int
        """
        self.ans = 1
        self.max_depth(root)
        return self.ans - 1

官方代码稍微简洁一些，写的方式感觉更好。

class Solution(object):
    def diameterOfBinaryTree(self, root):
        self.ans = 1
        def depth(node):
            # 访问到空节点了，返回0
            if not node: return 0
            # 左儿子为根的子树的深度
            L = depth(node.left)
            # 右儿子为根的子树的深度
            R = depth(node.right)
            # 计算d_node即L+R+1 并更新ans
            self.ans = max(self.ans, L+R+1)
            # 返回该节点为根的子树的深度
            return max(L, R) + 1

        depth(root)
        return self.ans - 1

作者：LeetCode-Solution
链接：https://leetcode-cn.com/problems/diameter-of-binary-tree/solution/er-cha-shu-de-zhi-jing-by-leetcode-solution/
来源：力扣（LeetCode）
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。