给定一个非空特殊的二叉树,每个节点都是正数,并且每个节点的子节点数量只能为 2
或 0
。如果一个节点有两个子节点的话,那么这个节点的值不大于它的子节点的值。
给出这样的一个二叉树,你需要输出所有节点中的第二小的值。如果第二小的值不存在的话,输出 -1 。
示例 1:
输入: 2 / \ 2 5 / \ 5 7 输出: 5 说明: 最小的值是 2 ,第二小的值是 5 。
示例 2:
输入: 2 / \ 2 2 输出: -1 说明: 最小的值是 2, 但是不存在第二小的值。
根据题目的描述,根节点都小于等于子节点的值,那么按照前序遍历取到第一个不等于root->val的值就是答案了,以为是bst的变形,但是不对,因为有如下情况:
所以按照前序遍历并不是由小到大,所以还是要找整棵树的第二小。
用一个cur来表示root节点的值,按照前序遍历搜索。(root节点表示根节点(永远不变),当前节点表示遍历到的当前位置节点(随遍历在改变))
1:如果当前节点的值比根节点root的值大,则有可能会成为第二小。
1则进一步判断cur中存的值是否等于根节点,如果是,则表明这是第一次替换cur的值,直接把当前节点的值替换给cur。
2如果cur中存的值不等于root节点,表明cur已经替换过了,则判断当前节点的值是否小于cur,如果小于则替换。
代码如下:
void findSecondMinimumValueCore(TreeNode* root, int &cur, TreeNode* root_val) { if (!root) { return; } if (root->val > root_val->val) { if (cur == root_val->val) { cur = root->val; } else { if (root->val < cur) { cur = root->val; } } } findSecondMinimumValueCore(root->left, cur, root_val); findSecondMinimumValueCore(root->right, cur, root_val); } int findSecondMinimumValue(TreeNode* root) { if (!root) { return -1; } int cur = root->val; bool flag = false; findSecondMinimumValueCore(root, cur, root); if (root->val != cur) { return cur; } return -1; }