671. 二叉树中第二小的节点
题目
给定一个非空特殊的二叉树,每个节点都是正数,并且每个节点的子节点数量只能为 2 或 0。如果一个节点有两个子节点的话,那么该节点的值等于两个子节点中较小的一个。
更正式地说,root.val = min(root.left.val, root.right.val) 总成立。
给出这样的一个二叉树,你需要输出所有节点中的第二小的值。如果第二小的值不存在的话,输出 -1 。
解题思路
该题很容易想到遍历树,然后用set来保存。取set中第下标1即是第二小的值。要注意使用treeSet<>
,因为hashSet<>
不保证有序。
代码
class Solution {
Set<Integer> set = new TreeSet<>();
public int findSecondMinimumValue(TreeNode root) {
if (root == null) {
return -1;
}
traverseTree(root);
int res = -1;
if (set.size() > 0) {
List<Integer> list = new ArrayList<>(set);
if(set.size()>=2){
res = list.get(1);
}
}
return res;
}
public void traverseTree(TreeNode root) {
if (root == null) {
return;
}
set.add(root.val);
traverseTree(root.left);
traverseTree(root.right);
}
}