【网易】二叉树权值最大的叶节点到权值最小的叶节点的距离

最新推荐文章于 2021-03-09 20:39:32 发布

原创

最新推荐文章于 2021-03-09 20:39:32 发布 · 978 阅读

标签

#网易 #dfs #二叉树 #python #算法

收录于

这是一篇关于如何在二叉树中寻找权值最大和最小的叶节点之间距离的博客。算法首先遍历树找到这两个节点，然后计算它们之间的路径长度减去公共祖先节点的距离。

【网易】二叉树权值最大的叶节点到权值最小的叶节点的距离

题目
- 输入
- 输出
- 解析
- 代码

题目

有一棵二叉树，树上每个点标有权值，权值各不相同，请设计一个算法算出权值最大的叶节点到权值最小的叶节点的距离。二叉树每条边的距离为1，一个节点经过多少条边到达另一个节点为这两个节点之间的距离。

给定二叉树的根节点root，请返回所求距离。

输入

[12,10,16,3,19,22,15,1,6,2,9,7,20,18,8,13,14,11,23,4,17,21,5,’#’]

输出

解析

先遍历找到最大的叶节点和最小的叶节点，然后减去路径上公共节点的距离。

代码

class TreeNode:
    def __init__(self, x):
        self.val = x
        self.left = None
        self.right = None

class Tree:
    def getDis(self, root):
        # write code here
        # 遍历所有节点，然后求距离
        maximum = [-0xffffff]
        minimum = [0xffffff]
        res = []
        self.dfs(root,[],maximum,minimum,res)
        list1 = []
        list2 = []
        for i in res:
            if maximum[0