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

80 阅读2分钟

这是我参与8月更文挑战的第11天,活动详情查看:8月更文挑战

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,那么我们可以将 res 的初始值置为 null。在遍历的过程中,如果当前节点的值严格大于 根节点的值 的节点时,那么只要 res 的值为 null 或者当前节点的值严格小于 res,我们就需要对res 进行更新。

此外,如果当前节点的值大于等于 res,那么根据「思路」部分,以当前节点为根的子树中所有节点的值都大于等于res,我们就直接回溯,无需对该子树进行遍历。这样做可以省去不必要的遍历过程。

代码

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    Integer res=null;
    public int findSecondMinimumValue(TreeNode root) {
        dfsFindSecondMinimumValue(root, root.val);
        return res==null?-1:res;
    }
    public void dfsFindSecondMinimumValue(TreeNode root,int tar) {

        if (root==null)  return;
        if(root.val>tar)
        {
            res=res==null?root.val:Math.min(res, root.val);
        }
        dfsFindSecondMinimumValue(root.left,tar);
        dfsFindSecondMinimumValue(root.right,tar);
    }
}