Appearance
230. 二叉搜索树中第K小的元素
题目描述
给定一个二叉搜索树的根节点 root ,和一个整数 k ,请你设计一个算法查找其中第 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 小的值,你将如何优化算法?
难度: Medium
题解代码
javascript
/**
* 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) {
const ret = []
let stack = []
let cur = root
while(cur || stack.length) {
while(cur) {
stack.push(cur)
cur = cur.left
}
cur = stack.pop()
ret.push(cur.val)
if (ret.length === k) return cur.val
cur = cur.right
}
};