leetcode(32)_671_easy_二叉树中第二小的节点_python

最新推荐文章于 2022-07-04 18:34:32 发布

m_a_rookie

最新推荐文章于 2022-07-04 18:34:32 发布

阅读量115

点赞数

分类专栏： python_leetcode 文章标签：二叉树 leetcode 递归算法

本文链接：https://blog.csdn.net/qq_40137656/article/details/119137916

版权

python_leetcode 专栏收录该内容

109 篇文章 0 订阅

订阅专栏

二叉树中第二小的节点

题目描述：
给定一个非空特殊的二叉树，每个节点都是正数，并且每个节点的子节点数量只能为 2 或 0。如果一个节点有两个子节点的话，那么该节点的值等于两个子节点中较小的一个。
更正式地说，root.val = min(root.left.val, root.right.val) 总成立。
给出这样的一个二叉树，你需要输出所有节点中的第二小的值。如果第二小的值不存在的话，输出 -1 。
示例：

输入：root = [2,2,5,null,null,5,7]
输出：5
解释：最小的值是 2 ，第二小的值是 5 。
提示：

树中节点数目在范围 [1, 25] 内
1 <= Node.val <= 231 - 1
对于树中每个节点 root.val == min(root.left.val, root.right.val)

解法

本题主要是要理清思路，最简单的将各节点的 val 存到集合中再排序，自然也不是最优解。结合根节点的值是孩子节点的较小值这一特殊性质，我们只需要看左子树和右子树中有没有比root.val还要大的值（如果有多个，取最小），如果都没有，那就返回 -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 findSecondMinimumValue(self, root: TreeNode) -> int:
        return self.dfs(root, root.val)
    
    def dfs(self, root, val):
        if not root:
            return -1
        if root.val > val:
            return root.val
        left_bigger = self.dfs(root.left, val)
        right_bigger = self.dfs(root.right, val)
        if left_bigger == -1 or right_bigger == -1:
            return max(left_bigger, right_bigger)
        return min(left_bigger, right_bigger)