LeetCode 671. 二叉树中第二小的节点（C++）

最新推荐文章于 2023-01-15 00:19:29 发布

原创最新推荐文章于 2023-01-15 00:19:29 发布 · 153 阅读

0 ·

CC 4.0 BY-SA版权

LeetCode 专栏收录该内容

713 篇文章

订阅专栏

本文探讨了在一个特殊二叉树中查找第二小数值的方法，该树的节点值均为正数，且每个节点要么没有子节点，要么有两个子节点。通过深度优先搜索遍历树的所有节点并排序，从而找出仅次于根节点值的最小值，若不存在则返回-1。

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

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

示例 1:

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

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

示例 2:

输入: 
    2
   / \
  2   2

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

C++

/**
 * 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 DFS(TreeNode* node, vector<int>& tmp)
    {
        if(node)
        {
            tmp.push_back(node->val);
            DFS(node->left,tmp);
            DFS(node->right,tmp);            
        }
    }
    

    int findSecondMinimumValue(TreeNode* root) 
    {
        vector<int> tmp;
        DFS(root->left,tmp);
        DFS(root->right,tmp);
        sort(tmp.begin(),tmp.end());
        int res=-1;
        if(tmp.size())
        {
            for(int i=0;i<tmp.size();i++)
            {
                if(tmp[i]>root->val)
                {
                    res=tmp[i];
                    break;
                }
            }
        }
        return res;
    }
};