nowcoder 树上最远距离

题目

从二叉树的节点A出发，可以向上或者向下走，但沿途的节点只能经过一次，当到达节点B时，路径上的节点数叫作A到B的距离。对于给定的一棵二叉树，求整棵树上节点间的最大距离。
给定一个二叉树的头结点root，请返回最大距离。保证点数大于等于2小于等于500.

思路

递归遍历左子树和右子树最大距离，更新全局最大距离。

代码

# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None
class LongestDistance:
    def __init__(self):
        self.max_distance = 0
    def find(self, root):
        if not root:
            return 0
        left_dis = self.find(root.left)
        right_dis = self.find(root.right)
        self.max_distance = max(self.max_distance, left_dis + right_dis + 1)
        return max(left_dis, right_dis) + 1
    def findLongest(self, root):
        # write code here
        self.find(root)
        return self.max_distance