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

题目来源

• leetcode：671. 二叉树中第二小的节点

题目描述

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 findSecondMinimumValue(TreeNode* root){

    }
};

题目解析

题意

这道题让我们找二叉树中的第二小的结点值，并且给该二叉树做了一些限制，比如对于任意一个结点，要么其没有子结点，要么就同时有两个子结点，而且父结点值是子结点值中较小的那个，当然两个子结点值可以相等。

分析

• 根据性质可以推导出，根节点一定是最小的那个节点，因此我们找出第二小的节点即可。
• 怎么办？
  • 令first = root.val， second = INT_MAX
  • 然后对根节点调用递归函数
    • 如果当前节点为空，直接返回
    • 如果当前节点孩子不等于first，说明其一定比first要大，那么我们看是否比second小，小的话就更新second
    • 然后对只有节点分别调用递归函数即可

class Solution {
public:
    int findSecondMinimumValue(TreeNode* root) {
        int first = root->val, second = INT_MAX;
        helper(root, first, second);
        return (second == first || second == INT_MAX) ? -1 : second;
    }
    void helper(TreeNode* node, int& first, int& second) {
        if (!node) return;
        if (node->val != first && node->val < second) {
            second = node->val;
        }
        helper(node->left, first, second);
        helper(node->right, first, second);
    }
};

思路二

递归找到不等于根结点大的值，不用再遍历当前节点下面的子节点，因为子节点的值不可能比它小。

class Solution {
private:


private:
    int helper(TreeNode* root, int val){
        if(root == NULL){
            return -1;
        }
        
        // 找到了，就返回
        if(root->val != val){
            return root->val;
        }
        
        // 找不到就去左右子树中找
        int l = helper(root->left, val);
        int r = helper(root->right, val);
        if(l == -1){
            return r;
        }
        if(r == -1){
            return l;
        }

        return std::min(l, r);
    }
public:
    int findSecondMinimumValue(TreeNode* root) {
        if(root == NULL){
            return -1;
        }

        return helper(root, root->val);
    }
};