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

36 阅读1分钟

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

一、题目描述:

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)

二、思路分析:

递归三部曲:

  1. 确定函数参数和返回值,该题中参数为节点,和节点值,往下递归时所有的子节点都跟该节点比较,返回值是-1或者第一个比根节点大的节点值;
  2. 确定终止条件,当遍历到叶节点时终止;
  3. 确定单层递归的逻辑,单层要么找到比节点大的值,要么为空返回-1,要么左右都找到返回较小的那个,要么只找到一边,另一边为空返回较大的那个即可

三、AC 代码:

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    int dfs(TreeNode* node, int val) {
        if(node == nullptr) return -1;
        if(node->val >  val) {
            return node->val;
        }
        int L = dfs(node->left, val);
        int R = dfs(node->right, val);
        if(L > 0 && R > 0) {
            return min(L, R);
        }
        else{
            return max(L, R);
        }
    }
    int findSecondMinimumValue(TreeNode* root) {
        return dfs(root, root->val);
    }
};