leetcode 671. 二叉树中第二小的节点(python)
程序员文章站
2024-02-18 10:58:52
...
给定一个非空特殊的二叉树,每个节点都是正数,并且每个节点的子节点数量只能为 2
或 0
。如果一个节点有两个子节点的话,那么这个节点的值不大于它的子节点的值。
给出这样的一个二叉树,你需要输出所有节点中的第二小的值。如果第二小的值不存在的话,输出 -1 。
示例 1:
输入: 2 / \ 2 5 / \ 5 7 输出: 5 说明: 最小的值是 2 ,第二小的值是 5 。
示例 2:
输入: 2 / \ 2 2 输出: -1 说明: 最小的值是 2, 但是不存在第二小的值。
# Definition for a binary tree node.
# class TreeNode(object):
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution(object):
def findSecondMinimumValue(self, root):
"""
:type root: TreeNode
:rtype: int
"""
s = set()
queue = [root]
while queue and queue[0]:
node = queue.pop(0)
s.add(node.val)
if node.left:
queue.append(node.left)
if node.right:
queue.append(node.right)
if len(s) < 2:
return -1
else:
s.remove(min(s))
return min(s)
推荐阅读