【LeetCode刷题】二叉树中第二小的节点


题目链接:二叉树中第二小的节点

题目

给定一个非空特殊的二叉树,每个节点都是正数,并且每个节点的子节点数量只能为 2 或 0。如果一个节点有两个子节点的话,那么该节点的值等于两个子节点中较小的一个。

更正式地说,root.val = min(root.left.val, root.right.val) 总成立。

给出这样的一个二叉树,你需要输出所有节点中的第二小的值。如果第二小的值不存在的话,输出 -1 。

示例1:

输入:root = [2,2,5,null,null,5,7]
输出:5
解释:最小的值是 2 ,第二小的值是 5 。

示例2:

输入:root = [2,2,2]
输出:-1
解释:最小的值是 2, 但是不存在第二小的值。

解题

我的想法很简单,遍历二叉树,找到最小的节点,同时保存遍历序列,然后对遍历序列进行排序,再与最小值进行比对,找到第二小的值。

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    int res = Integer.MAX_VALUE;
    List<Integer> vals = new ArrayList();
    public void findMin(TreeNode T) {
        if(T != null) {
            vals.add(T.val);
            if(T.val < res) {
                res = T.val;
            }

            findMin(T.left);
            findMin(T.right);
        }
    }

    public int findSecondMinimumValue(TreeNode root) {
        findMin(root);

        Collections.sort(vals);
        for(int a : vals) {
            if(a > res) {
                res = a;
                break;  // 有序,存在比 res(最小值)大的数,说明找到了第二小的数,退出循环。
            }
        }
        
        // vals经过排序,第一个数一定是最小值,如果res和最小值相等,说明不存在第二小值,则返回-1
        return res == vals.get(0) ? -1 : res;
    }
}

改进

感觉有改进空间,奈何今天时间太紧,空先占着,哪天想起来再更新。

声明:楓の街|版权所有,违者必究|如未注明,均为原创|本网站采用BY-NC-SA协议进行授权

转载:转载请注明原文链接 - 【LeetCode刷题】二叉树中第二小的节点


Just For Fun...