LeetCode--230.二叉搜索树中第K小的元素(JavaScript)

给定一个二叉搜索树,编写一个函数 kthSmallest 来查找其中第 k 个最小的元素。

说明:
你可以假设 k 总是有效的,1 ≤ k ≤ 二叉搜索树元素个数。

示例 1:

输入: root = [3,1,4,null,2], k = 1
   3
  / \
 1   4
  \
   2
输出: 1

示例 2:

输入: root = [5,3,6,2,4,null,null,1], k = 3
       5
      / \
     3   6
    / \
   2   4
  /
 1
输出: 3

进阶:
如果二叉搜索树经常被修改(插入/删除操作)并且你需要频繁地查找第 k 小的值,你将如何优化 kthSmallest 函数?


思路

简单版:

使用中序遍历,转换为数组,此数组便是递增数组,返回第 k-1 个元素即可。

/** * Definition for a binary tree node. * function TreeNode(val) { * this.val = val; * this.left = this.right = null; * } */
/** * @param {TreeNode} root * @param {number} k * @return {number} */
var kthSmallest = function(root, k) {
  let res = []
  const inorder = (root) => {
    if (root) {
      inorder(root.left);
      res.push(root.val);
      inorder(root.right);
    }
  }
  inorder(root);
  return res[k-1]
};

进阶版:

采用递归的方法,先求出根节点左子树的所有节点数 leftNum,

  const count = (root) => {
    if (!root) {
      return 0;
    } else {
      return 1 + count(root.left) + count(root.right);
    }
  }
  1. leftNum < k - 1 , 说明第 k 个元素在根节点的右子树中,递归返回 kthSmallest(root.right, k - leftNum - 1)
  2. leftNum = k - 1,说明第 k 个元素刚好是根节点,返回根节点的值即可
  3. leftNum > k - 1,说明第 k 个元素在根节点的左子树中,递归返回kthSmallest(root.left, k)

使用递归的好处是不用额外空间,并且当树频繁修改时,也可以很好的适应。
全部代码:

/** * Definition for a binary tree node. * function TreeNode(val) { * this.val = val; * this.left = this.right = null; * } */
/** * @param {TreeNode} root * @param {number} k * @return {number} */
var kthSmallest = function(root, k) {
  const count = (root) => {
    if (!root) {
      return 0;
    } else {
      return 1 + count(root.left) + count(root.right);
    }
  }
  
  let leftNum = count(root.left);
  if (leftNum === k - 1) {
    return root.val;
  } else if (leftNum > k - 1) {
    return kthSmallest(root.left, k)
  } else {
    return kthSmallest(root.right, k - leftNum - 1)
  }
};
全部评论

相关推荐

刚刷到字节跳动官方发的消息,确实被这波阵仗吓了一跳。在大家还在纠结今年行情是不是又“寒冬”的时候,字节直接甩出了史上规模最大的转正实习计划——ByteIntern。咱们直接看几个最硬的数,别被花里胡哨的宣传词绕晕了。首先是“量大”。全球招7000多人是什么概念?这几乎是把很多中型互联网公司的总人数都给招进来了。最关键的是,这次的资源分配非常精准:研发岗给了4800多个Offer,占比直接超过六成。说白了,字节今年还是要死磕技术,尤其是产品和AI领域,这对于咱们写代码的同学来说,绝对是今年最厚的一块肥肉。其次是大家最关心的“转正率”。官方直接白纸黑字写了:整体转正率超过50%。这意味着只要你进去了,不划水、正常干,每两个人里就有一个能直接拿校招Offer。对于2027届(2026年9月到2027年8月毕业)的同学来说,这不仅是实习,这简直就是通往大厂的快捷通道。不过,我也得泼盆冷水。坑位多,不代表门槛低。字节的实习面试出了名的爱考算法和工程实操,尤其是今年重点倾斜AI方向,如果你简历里有和AI相关的项目,优势还是有的。而且,转正率50%也意味着剩下那50%的人是陪跑的,进去之后的考核压力肯定不小。一句话总结:&nbsp;27届的兄弟们,别犹豫了。今年字节这是铁了心要抢提前批的人才,现在投递就是占坑。与其等到明年秋招去千军万马挤独木桥,不如现在进去先占个工位,把转正名额攥在手里。
喵_coding:别逗了 50%转正率 仔细想想 就是转正与不转正
字节7000实习来了,你...
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务