671. 二叉树中第二小的节点 力扣(简单) auto循环set
题目描述:
给定一个非空特殊的二叉树,每个节点都是正数,并且每个节点的子节点数量只能为 2 或 0。如果一个节点有两个子节点的话,那么该节点的值等于两个子节点中较小的一个。
更正式地说,root.val = min(root.left.val, root.right.val) 总成立。
给出这样的一个二叉树,你需要输出所有节点中的第二小的值。如果第二小的值不存在的话,输出 -1 。
示例 1:
输入:root = [2,2,5,null,null,5,7]
输出:5
解释:最小的值是 2 ,第二小的值是 5 。
题源:https://leetcode-cn.com/problems/second-minimum-node-in-a-binary-tree/
代码:
/** * 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) { int maxval=root->val; queue<TreeNode* > Q; Q.push(root); set<int> s; while(!Q.empty()) { TreeNode* p=Q.front(); Q.pop(); if(p->left!=NULL) { Q.push(p->left); s.insert(p->left->val); } if(p->right!=NULL) { Q.push(p->right); s.insert(p->right->val); } } int res=-1; for(auto i: s) if( i > maxval) {res=i; break;} return res; } };