链接:230. 二叉搜索树中第 K 小的元素 - 力扣(LeetCode)
题目:
给定一个二叉搜索树的根节点
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
思路:
没看清题目是搜索二叉树,误以为是二叉树,所以用的是层次遍历,时间较慢。
代码:
/**
* 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)
* }
*/
/**
* @param {TreeNode} root
* @param {number} k
* @return {number}
*/
var kthSmallest = function(root, k) {
// 用arr存前k最小值
let arr = new Array(k).fill(-1)
let rt = [root]
// 层次遍历
while(rt.length != 0){
let num = rt[0].val
rt[0].left==null?null:rt.push(rt[0].left)
rt[0].right==null?null:rt.push(rt[0].right)
rt.shift()
if(arr[k-1]!=-1&&arr[k-1]<=num) continue
let i = k - 2
// 类似于插入排序
for( ; i>=0 ; i-- ){
if(arr[i]==-1) continue
if(arr[i]<=num){
arr[i+1] = num
break
}else {
arr[i+1] = arr[i]
}
}
if(i == -1) arr[i+1] = num
}
//返回数组最后一个第k小的值
return arr[k-1]
};