注:本文的代码实现使用的是 JS(JavaScript),为前端中想使用JS练习算法和数据结构的小伙伴提供解题思路。
描述
给定一个二叉搜索树的根节点 root ,和一个整数 k ,请你设计一个算法查找其中第 k 个最小元素(从 1 开始计数)。
示例:
输入:root = [3,1,4,null,2], k = 1
输出:1
输入:root = [5,3,6,2,4,null,null,1], k = 3
输出:3
解题思路
对于二叉搜索树来说,有一以下性质:
- 节点的左子树只包含小于当前节点的数。
- 节点的右子树只包含大于当前节点的数。
- 所有左子树和右子树自身必须也是二叉搜索树。
因此,我们可以推断出,二叉搜索树的先序遍历后得到的结果一定是递增的。对于这道题,我们可以先进行先序遍历,然后取下标为k-1
的那个元素即可。
var kthSmallest = function(root, k) {
// 定义一个数组,用于存放先序遍历后的结果
let nums = []
// 用于先序遍历的递归函数
var preOrder = root => {
if(root === null) return
if(root.left) preOrder(root.left)
nums.push(root.val)
if(root.right) preOrder(root.right)
}
preOrder(root)
// 返回第 k-1 个值,即第 k 大的主数
return nums[k-1]
};