leetcode 671. 二叉树中第二小的节点(Second Minimum Node In a Binary Tree)

目录

• 题目描述：
• 示例 1:
• 示例 2:
• 解法：

题目描述：

给定一个非空特殊的二叉树，每个节点都是正数，并且每个节点的子节点数量只能为 2 或 0。如果一个节点有两个子节点的话，那么这个节点的值不大于它的子节点的值。

给出这样的一个二叉树，你需要输出所有节点中的第二小的值。如果第二小的值不存在的话，输出 -1 。

示例 1:

输入: 
        2
       / \
      2   5
         / \
        5   7

输出: 5
说明: 最小的值是 2 ，第二小的值是 5 。

示例 2:

输入: 
        2
       / \
      2   2

输出: -1
说明: 最小的值是 2, 但是不存在第二小的值。

---

解法：

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    void findValue(TreeNode* root, int& min1, int& min2){
        if(root){
            if(!root->left && !root->right){
                min1 = root->val;
                min2 = -1;
            }else{
                int lmin1 = -1, lmin2 = -1;
                int rmin1 = -1, rmin2 = -1;
                findValue(root->left, lmin1, lmin2);
                findValue(root->right, rmin1, rmin2);
                unordered_set<int> st = {root->val, lmin1, lmin2, rmin1, rmin2};
                vector<int> vals(st.begin(), st.end());
                sort(vals.begin(), vals.end());
                int sz = vals.size();
                if(sz == 1){
                    min1 = vals[0];
                    min2 = -1;
                }else if(sz == 2){
                    min1 = vals[1];
                    min2 = -1;
                }else{
                    if(vals[0] == -1){
                        min1 = vals[1];
                        min2 = vals[2];
                    }else{
                        min1 = vals[0];
                        min2 = vals[1];
                    }
                }
            }
        }
    }
    
    int findSecondMinimumValue(TreeNode* root) {
        int min1 = -1, min2 = -1;
        findValue(root, min1, min2);
        return min2;
    }
};