543. 二叉树的直径

543. 二叉树的直径

原始题目链接：https://leetcode-cn.com/problems/diameter-of-binary-tree/
给定一棵二叉树，你需要计算它的直径长度。一棵二叉树的直径长度是任意两个结点路径长度中的最大值。这条路径可能穿过也可能不穿过根结点。

解题思路：

题目出现任意两个节点，会联想到在遍历的过程中需要比较并更新最大值，或者使用双递归的方式解题。遍历二叉树的方式根据题意，使用后序遍历方式自底向上来计算比较方便。计算树中节点的高度，可以用递归的方式来解题，当前节点的高度等于其左右子树的高度+1，加1是其节点本身的高度。每次比较并更新最大值的时候，当前计算的任意两个节点路径长度的最大值等于(左子树 + 右子树)的返回值之和。具体实现看代码及注释。

代码实现：

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    def __init__(self):
        # 定义最终答案变量，初始化为0
        self.ans = 0
    
    # 定义一个递归函数：计算当前节点的高度，当前节点的高度等于其左右子树中的最大值+1
    # 递归函数的参数是当前节点，递归函数的结束条件是空节点返回0
    # 递归函数的等价关系式是传递当前节点的左、右节点
    # 当计算当前节点的高度时，需要知道其左右子树的高度，所以先递归遍历到最底层，从底向上计算，所以使用后序遍历的方式
    def dfs(self, root):
        # 递归结束条件
        if root is None:
            return 0
        L = self.dfs(root.left)
        R = self.dfs(root.right)
        # 因为题目要求是任意两个节点，可能不过最上面的根节点，所以每次需要比较目前最大值和目前计算得到的一棵二叉树的直径长度(L + R)
        # 更新答案
        self.ans = max(self.ans, L + R)
        # 每次返回当前节点的最大子树的高度，再加1得到当前节点的高度
        return max(L, R) + 1

    def diameterOfBinaryTree(self, root: TreeNode) -> int:
        if root is None:
            return 0
        self.dfs(root)
        return self.ans

参考文献：
https://leetcode-cn.com/problems/diameter-of-binary-tree/solution/hot-100-9er-cha-shu-de-zhi-jing-python3-di-gui-ye-/