js.二叉搜索树中第K小的元素

链接：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].valrt[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) continuelet i = k - 2// 类似于插入排序for( ; i>=0 ; i-- ){if(arr[i]==-1) continueif(arr[i]<=num){arr[i+1] = numbreak}else {arr[i+1] = arr[i]}}if(i == -1) arr[i+1] = num}//返回数组最后一个第k小的值return arr[k-1]
};