LeetCode——671. 二叉樹中第二小的節點

講個笑話哈哈發表於2020-11-27

題目描述:

給定一個非空特殊的二叉樹,每個節點都是正數,並且每個節點的子節點數量只能為 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, 但是不存在第二小的值。

程式碼如下:

/**
 * 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);
    }
};

執行結果:

在這裡插入圖片描述

相關文章