leetcode543.二叉树的直径 Python

题目：

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

示例：

给定二叉树

      1
     / \
    2   3
   / \     
  4   5

返回 3, 它的长度是路径 [4,2,1,3] 或者 [5,2,1,3]。
注意：两结点之间的路径长度是以它们之间边的数目表示。

思路：

递归，深度优先遍历
求直径就是求两个节点之间的最大距离，结果应该是从根节点出发，左子树的最大路径长度和右子树的最大路径之和，在递归过程中要维护这个最大值，在递归函数中返回的是一棵子树路径长度的最大值，整体来看，由根节点出发，左子树作为参数传入递归函数，最终返回的是左子树中的最大路径长度，由根节点出发，右子树作为参数传递给递归函数，最终返回的是右子树中的最大路径长度，要求的结果是这两个长度相加，最后返回维护的最大值，即两个节点之间的最大距离。

# Definition for a binary tree node.
# class TreeNode(object):
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution(object):
    def diameterOfBinaryTree(self, root):
        """
        :type root: TreeNode
        :rtype: int
        """
        # 递归，深度优先遍历
        self.ans = 0
        def dfs(root):
            if not root:
                return 0
            left = dfs(root.left)
            right = dfs(root.right)
            self.ans = max(self.ans,left + right)
            return max(left,right) + 1
        dfs(root)
        return self.ans