leetcode 671. Second Minimum Node In a Binary Tree 第二小数字 + 深度优先遍历DFS

逃离我推掉我的手 2022-06-03 06:53 133阅读 0赞

Given a non-empty special binary tree consisting of nodes with the non-negative value, where each node in this tree has exactly two or zero sub-node. If the node has two sub-nodes, then this node’s value is the smaller value among its two sub-nodes.

Given such a binary tree, you need to output the second minimum value in the set made of all the nodes’ value in the whole tree.

If no such second minimum value exists, output -1 instead.

Example 1:  
Input:  
2  
/ \\  
2 5  
/ \\  
5 7

Output: 5  
Explanation: The smallest value is 2, the second smallest value is 5.  
Example 2:  
Input:  
2  
/ \\  
2 2

Output: -1  
Explanation: The smallest value is 2, but there isn’t any second smallest value.

这道题题意很简单，本题描述的二叉树假如只可能存在两个子节点或者0个，假如存在两个，那么本节点就是这三个结点中最小的值，那么可以得到的信息就是root结点就是最小的值，那么遍历一次即可找到第二小的值

代码如下：

#include <iostream>
    #include <vector>
    #include <map>
    #include <set>
    #include <queue>
    #include <stack>
    #include <string>
    #include <climits>
    #include <algorithm>
    #include <sstream>
    #include <functional>
    #include <bitset>
    #include <numeric>
    #include <cmath>
    #include <regex>
    
    using namespace std;
    
    
    /*
    struct TreeNode 
    {
         int val;
         TreeNode *left;
         TreeNode *right;
         TreeNode(int x) : val(x), left(NULL), right(NULL) {}
    };
    */
    
    class Solution 
    {
    public:
        int findSecondMinimumValue(TreeNode* root) 
        {
            if (root == NULL)
                return -1;
    
            int min = root->val;
            int secMin = INT_MAX;
            dfs(root,min,secMin);
            if (secMin == INT_MAX )
                return -1;
            else
                return secMin;
        }
    
        void dfs(TreeNode* root,int& min,int& secMin)
        {
            if (root != NULL)
            {
                if (root->val != min && root->val < secMin)
                    secMin = root->val;
                dfs(root->left, min, secMin);
                dfs(root->right, min, secMin);
            }
        }
    };