2021-12-03 671. 二叉树中第二小的节点【Easy】

2021-12-03  本文已影响0人  JackHCC

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

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

给出这样的一个二叉树,你需要输出所有节点中的第二小的值。如果第二小的值不存在的话,输出 -1 。

示例 1:


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

示例 2:


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

提示:

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

方法一:BFS与DFS

# 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 BFS(root)
        return DFS(root, root.val)
        

def BFS(root):
    s1 = root.val
    s2 = float('inf')
    found = False

    q = []
    q.append(root)

    while len(q) != 0:
        node = q[0]
        q.remove(q[0])

        if node.val > s1 and node.val < s2:
            s2 = node.val
            found = True
            continue
        
        if node.left ==  None: continue
        q.append(node.left)
        q.append(node.right)

    return s2 if found else -1



def DFS(root, s1):
    if root == None: return -1

    if root.val > s1: return root.val
    sl = DFS(root.left, s1)
    sr = DFS(root.right, s1)
    if sl == -1: return sr
    if sr == -1: return sl

    return min(sl, sr)
上一篇 下一篇

猜你喜欢

热点阅读