LeetCode543. 二叉树的直径

1、题目描述

https://leetcode-cn.com/problems/diameter-of-binary-tree/

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

2、代码详解

红线的直径是8（边数）

任意一个结点，都要记录以此结点为根的直径情况：左子树高度+右子树高度

需要一个值来保存我这个每次比较更新的最大直径值，用self.ans = 1来初始化节点数（边数 = 节点数 - 1）
在每次获得一个节点的左子树和右子树的值的时候，都需要比较一下self.ans和左子树高度+右子树高度+1的大小，把更大的保存下来
然后如何求左子树和右子树的高度呢，那就是经典的递归求高度问题：max(depth(root.left), depth(root.right))+1

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

class Solution(object):
    def diameterOfBinaryTree(self, root):
        """
        :type root: TreeNode
        :rtype: int
        """
        self.ans = 1  # 节点数
        def depth(root):
            if not root:
                return 0
            L = depth(root.left)
            R = depth(root.right)
            self.ans = max(self.ans, L + R + 1)
            return max(L, R) + 1
        depth(root)
        return self.ans - 1  # 边数