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

48 阅读1分钟

开启掘金成长之旅!这是我参与「掘金日新计划 · 12 月更文挑战」的第26天,点击查看活动详情

一、题目描述:

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)

二、思路分析:

要找第二小的值,那么肯定要完整遍历树

  • 直接用最简单的层序遍历整个树,将树各节点值保存在一个数组中
  • 将保存结果的数组对值进行排序,从小到大
  • 那么最小的肯定是第一个值
  • 再for循环,等于最小的继续循环,那后续第一个不等于就是第二小的了
  • 函数最后返回-1,就是以上情况都没有

三、AC 代码:

/**
 * Definition for a binary tree node.
 * type TreeNode struct {
 *     Val int
 *     Left *TreeNode
 *     Right *TreeNode
 * }
 */
func findSecondMinimumValue(root *TreeNode) int {
    if root == nil {
        return -1
    }
    queue := []*TreeNode{root}
    values := []int{}
    for len(queue) > 0 {
        curr := queue[0]
        values = append(values, curr.Val)   
        queue = queue[1:]
        if curr.Left != nil {
            queue = append(queue, curr.Left)
        }
        if curr.Right != nil {
            queue = append(queue, curr.Right)
        }
    }
    sort.Ints(values)
    min := values[0]
    for _, v := range values {
        if v == min {
            continue
        }
        return v
    }
    return -1
}

范文参考:

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