力扣-二叉树中第k小的元素 med 🤭

Offer 驾到，掘友接招！我正在参与2022春招打卡活动，点击查看活动详情。

前言

数据结构与算法属于开发人员的内功，不管前端技术怎么变，框架怎么更新，版本怎么迭代，它终究是不变的内容。 始终记得在参加字节青训营的时候，月影老师说过的一句话，不要问前端学不学算法。计算机学科的每一位都有必要了解算法，有写出高质量代码的潜意识。

一、问题描述

给定一个二叉搜索树的根节点 root ，和一个整数 k ，请你设计一个算法查找其中第 k 个最小元素（从 1 开始计数）。

示例 1：

输入：root = [3,1,4,null,2], k = 1
输出：1

示例 2：

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

提示：

• 树中的节点数为 n 。
• 1 <= k <= n <= 104
• 0 <= Node.val <= 104

进阶：如果二叉搜索树经常被修改（插入/删除操作）并且你需要频繁地查找第 k 小的值，你将如何优化算法？

二、思路讲解

既然是二叉搜索树的话，不难想到二叉搜索树的中序遍历结果就是递增的。再次基础上我们只需要创建一个数组存储中序遍历结果，最后返回第k个元素即可。 空间复杂度 O（N）

var kthSmallest = function(root, k) {
   const arr = [] 
   const dfs = (root) => {
       if(!root) return  // 递归终止条件
       dfs(root.left) // 中序遍历
       arr.push(root.val)
       dfs(root.right)
   }
   dfs(root)
   return arr[k-1]
};

优化上述代码，既然数组中除了第k-1 个元素之外，其它元素都是无效的。那我们就可以使用一个count来记录当前是第几个元素，最后返回第k-1个元素即可。

var kthSmallest = function(root, k) {
   let count = 0 // 用于记录当前是第几个元素
   let res = 0 // 最终结果
   const dfs = (root) =>{
       if(!root) return
        dfs(root.left)
        if(count++==k-1){ // 每次判断完一个元素，不管是不是第k-1个元素，count++，避免后面的元素影响到结果
            res = root.val
        }
        dfs(root.right)
   }
   dfs(root)
   return res
};

三、测试结果

后续

• 地址: 二叉树中第k小的元素

好了，本篇 力扣-二叉树中第k小的元素到这里就结束了，我是邵小白，一个在前端领域摸爬滚打的大三学生，欢迎👍评论。