Leetcode-671. 二叉树中第二小的节点

给定一个非空特殊的二叉树,每个节点都是正数,并且每个节点的子节点数量只能为 2 或 0。如果一个节点有两个子节点的话,那么该节点的值等于两个子节点中较小的一个。

更正式地说,root.val = min(root.left.val, root.right.val) 总成立。

给出这样的一个二叉树,你需要输出所有节点中的第二小的值。如果第二小的值不存在的话,输出 -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 findSecondMinimumValue(self, root: TreeNode) -> int:
        mixnode = root.val
        ans = -1
        def find(root):
            nonlocal ans
            if not root:
                return
            # 排除ans已经有值,并且结点值大于等于ans的结点
            if ans != -1 and root.val >= ans:
                return
            # 处理第一次ans没有值和之后严格大于根节点的值
            if root.val > mixnode:
                ans = root.val
            
            find(root.left)
            find(root.right)

        find(root)
        return ans

方法2:

# 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:
        def find(node):
            if not node:
                return float('inf')
            if node.val != root.val:
                return node.val
            else:
                return min(find(node.left), find(node.right))
        x = find(root)
        if x == float('inf'):
            return -1
        else:
            return x


posted @ 2021-07-27 10:48  小Aer  阅读(2)  评论(0编辑  收藏  举报  来源