671.二叉树中第二小的节点
给定一个非空特殊的二叉树,每个节点都是正数,并且每个节点的子节点数量只能为 2 或 0。如果一个节点有两个子节点的话,那么这个节点的值不大于它的子节点的值。
给出这样的一个二叉树,你需要输出所有节点中的第二小的值。如果第二小的值不存在的话,输出 -1 。
示例 1:
输入:
2
/ \
2 5
/ \
5 7
输出: 5
说明: 最小的值是 2 ,第二小的值是 5 。
示例 2:
输入:
2
/ \
2 2
输出: -1
说明: 最小的值是 2, 但是不存在第二小的值。
分析: 用fir表示最小值,用sec表示第二小值,先序遍历每个节点,若小于等于fir则更新fir;若大于fir,小于sec则更新sec。结果要和INT_MAX比较,取最小值。要注意的是root节点为空或其左右孩子为空的情况。
/**
* 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:
int findSecondMinimumValue(TreeNode* root) {
if(root==NULL) return -1;
if(root->left==NULL&&root->right==NULL) return -1;
int fir=INT_MAX;
int sec=INT_MAX;
preord(root,fir,sec);
return sec==INT_MAX?-1:sec;
}
void preord(TreeNode* root,int &fir,int &sec)
{
if(root==NULL) return;
if(root->val<=fir) fir=root->val;
else if(root->val<sec) sec=root->val;
preord(root->left,fir,sec);
preord(root->right,fir,sec);
return;
}
};