leetcode算法671. 二叉树中第二小的节点

57 阅读1分钟

持续创作,加速成长!这是我参与「掘金日新计划 · 10 月更文挑战」的第28天,点击查看活动详情

一、题目描述:

671. 二叉树中第二小的节点 - 力扣(LeetCode)

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

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

给出这样的一个二叉树,你需要输出所有节点中的 第二小的值

如果第二小的值不存在的话,输出 -1

 

示例 1:

image.png

输入:root = [2,2,5,null,null,5,7]
输出:5
解释:最小的值是 2 ,第二小的值是 5

示例 2:

image.png

输入:root = [2,2,2]
输出:-1
解释:最小的值是 2, 但是不存在第二小的值。

 

提示:

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

二、思路分析:

父节点的值总是左右子节点中最小的 目标:找出第二小的值

思路:根节点一定是最小的,利用队列广度查找,如果该层中有比根节点大的,记录下来这个值,存在与根节点值相等的node,加到队列中,直到队列为空

三、AC 代码:

# 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 TreeNode:
    def __init__(self, val=0, left=None, right=None):
        self.val = val
        self.left = left
        self.right = right

class Solution:
    def __init__(self):
        self.min = 0
        # 假定不存在
        self.target = -1

    def findSecondMinimumValue(self, root: TreeNode) -> int:
        self.min = root.val
        level = deque([root])
        while level:
            for i in range(0, len(level)):
                node = level.popleft()
                # 找到了第一个比最小值大的
                if node.val > self.min and self.target == -1:
                    self.target = node.val

                # 找到了比最小值大,但是比已有的值更小一些的值
                elif self.target > node.val > self.min:
                    self.target = node.val

                # 等于最小值时,目标可能在子节点
                elif node.val == self.min and node.left:
                    level.append(node.left)
                    level.append(node.right)
        return self.target