671 二叉树中第二小的节点 简单
发布者:admin 发表于:442天前 阅读数:501 评论:0

671. 二叉树中第二小的节点 简单

给定一个非空特殊的二叉树,每个节点都是正数,并且每个节点的子节点数量只能为 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 <= 231 - 1

对于树中每个节点 root.val == min(root.left.val, root.right.val)

代码参考:

package main

import "fmt"

func main() {
    root := &TreeNode{Val: 5}
    root.Left = &TreeNode{Val: 5}
    root.Right = &TreeNode{Val: 6}
    fmt.Println(findSecondMinimumValue(root)) // 6
}

const MAX = 1 << 31

// 不动脑筋的做法
func findSecondMinimumValue(root *TreeNode) int {
    if root == nil {
        return -1
    }

    q := []*TreeNode{root}
    min := MAX
    for len(q) > 0 {
        count := len(q)
        for i := 0; i < count; i++ {
            cur := q[0]
            q = q[1:]
            if cur.Val != root.Val && cur.Val < min {
                min = cur.Val
            }
            if cur.Left != nil {
                q = append(q, cur.Left)
            }
            if cur.Right != nil {
                q = append(q, cur.Right)
            }
        }
    }
    if min != MAX && min != root.Val {
        return min
    }

    return -1
}