二叉树的直径

题目

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

示例 :

给定二叉树

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/diameter-of-binary-tree
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

解题思路

概念理解

• 路径长度
  一条路径的长度为该路径经过的结点数减一。
• 题目中的直径
  即求路径长度的最大值。

这条路径可能穿过也可能不穿过根结点 → 任意一条路径可由某结点为起点，再向下遍历其左孩子和右孩子得到。

设某结点的左孩子向下遍历经过最多的结点数为 l（以左孩子为根），其右儿子向下遍历经过最多的结点数为 r （以右孩子为根），则以该结点为起点的路径经过节点数的最大值即为 l + r + 1（再加上根节点） 。

题目所给函数名是直接得到结果，于是在解题时需要定义递归函数，记录某结点 node 为起点的路径经过节点数的最大值为 self.d，为全局变量，返回的结果为 self.d - 1。

递归函数返回该结点为根的子树的深度。先递归调用左孩子和右孩子分别求以它们为根的子树的深度 l 和 r ，则该结点为根的子树的深度为max(l,r)+1。

提交代码

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution:
    def diameterOfBinaryTree(self, root: TreeNode) -> int:
        self.d = 1
        def depth(node):
            if not node:
                return 0

            l = depth(node.left)
            r = depth(node.right)
            self.d = max(self.d, l+r+1)
            return max(l, r) + 1

        depth(root)
        return self.d - 1

总结

与二叉树的最大深度题目的递归原理相同，向下遍历每个结点的左右子树，此题中，路径长度与路径所经过的结点数相关，直径是任意两结点间的最长的路径长度。