JS 笔试题——二叉树中第二小的节点

2021-12-06  本文已影响0人  袭明_

给定一个非空特殊的二叉树,每个节点都是正数,并且每个节点的子节点数量只能为 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.
 * function TreeNode(val, left, right) {
 *     this.val = (val===undefined ? 0 : val)
 *     this.left = (left===undefined ? null : left)
 *     this.right = (right===undefined ? null : right)
 * }
 */

分析:

  1. 涉及到有序二叉树,尽量往回溯算法上靠;
  2. 根节点小于子节点;
  3. 只要记录比根节点大的子节点即可,如果遇到其他节点比当前 ans 值大,则终止,回溯即可;
  4. 如果当前节点值大于根节点的值,且没有比 ans 大,则刷新 ans ;

代码:

/**
* @param {TreeNode} root
* @return {number}
*/
var findSecondMinimumValue = function(root) {
   let ans = -1;
   const rootValue = root.val;
   const dfs = (node) => {
       if (node == null) {
           return;
       }
       if (ans !== -1 && node.val >= ans) {
           return;
       }
       if (node.val > rootValue) {
           ans = node.val
       }
       dfs(node.left);
       dfs(node.right);
   }
   dfs(root);
   return ans;
};

复杂度分析:

时间复杂度:O(n),其中 n 是二叉树中的节点个数。最多需要对整棵二叉树进行一次遍历。

空间复杂度:O(n)。使用深度优先搜索的方法进行遍历,需要使用的栈空间为 O(n)

上一篇 下一篇

猜你喜欢

热点阅读