（Python）LeetCode 543:二叉树的直径

题目

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

示例

给定二叉树

      1
     / \
    2   3
   / \     
  4   5    

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

说明

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

Python实现

思路（递归） 循环所有节点，求经过每个节点的最长路径长度——左、右子树深度-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.result = 0 #存储最大路径长度
        def treemax(tree): #循环tree得出经过该节点的最大路径长度
            if tree == None: #若tree为空树，直接返回0
                return 0
            l=treemax(tree.left) #获取tree左树深度-1
            r=treemax(tree.right) #获取tree右树深度-1
            #若经过tree的最大路径长度（（左树深度-1+右树深度-1）+1）比原记录的最大路径长度大，则替换result值
            self.result = max(l+r+1,self.result)
            return max(l,r)+1     #返回tree深度，便于treemax函数的递归
        if root == None:   # 避免直接直接求空树的最长路径
            return 0
        treemax(root)
        return self.result-1 #因最终所求最长路径长度左、右子树的深度均+1（即加根节点），故需要减去重复加的深度（即-1）

【甜点来袭1】内、外函数公用同一个变量result的方法——在内、外函数中通过self调用result变量，即self.result。