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

题目描述：

给定一个非空特殊的二叉树，每个节点都是正数，并且每个节点的子节点数量只能为 2 或 0。如果一个节点有两个子节点的话，那么该节点的值等于两个子节点中较小的一个。更正式地说，root.val = min(root.left.val, root.right.val) 总成立。给出这样的一个二叉树，你需要输出所有节点中的第二小的值。如果第二小的值不存在的话，输出 -1 。

提示：

树中节点数目在范围 [1, 25] 内
1 <= Node.val <= 2^31 - 1
对于树中每个节点 root.val == min(root.left.val, root.right.val)

示例 1：

输入：root = [2,2,5,null,null,5,7]
输出：5
解释：最小的值是 2 ，第二小的值是 5 。

示例 2：

输入：root = [2,2,2]
输出：-1
解释：最小的值是 2, 但是不存在第二小的值。

JAVA代码如下：

/**
 * 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 {
    public static int findSecondMinimumValue(TreeNode root) {
        if (root == null) {
            return -1;
        }
        int ans = root.val;
        ArrayList<Integer> arr = new ArrayList<>();
        pre(root, arr);
        HashSet<Integer> set = new HashSet<>();
        set.addAll(arr);
        int res = -1;
        if (set.size() == 1) {
            res = -1;
        } else {
            for (Integer n : set) {
                if (n > ans) {
                    res = n;
                    break;
                }
            }
        }
        return res;
    }

    public static void pre(TreeNode root, ArrayList<Integer> arr) {
        if (root == null) {
            return;
        }
        arr.add(root.val);
        pre(root.left, arr);
        pre(root.right, arr);
    }
}

C++代码如下：

/**
 * 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 findSecondMinimumValue(TreeNode* root) {
        vector<int>v;
        if(root==nullptr){
            return -1;
        }
        traverse(root,v);
        sort(v.begin(),v.end());
        int min=v[0];
        for(int i=1;i<v.size();i++){
            if(v[i]!=v[0]){
                min=v[i];
                break;
            }
        }
        if(min==v[0]){
            return -1;
        }
        else{
            return min;
        }
    }
    void traverse(TreeNode*root,vector<int>&v)
    {
        if(root==nullptr){
            return;
        }
        v.push_back(root->val);
        traverse(root->left,v);
        traverse(root->right,v);
    }
};

执行结果：