题目描述:给定一个二叉搜索树,编写一个函数 kthSmallest 来查找其中第 k 个最小的元素。
说明:你可以假设 k 总是有效的,1 ≤ k ≤ 二叉搜索树元素个数
。
根据二叉搜索树(BST)的性质:节点的权重大于左节点的权重,大于右节点的权重。因此,对 BST 进行中序遍历,就是按从小到大的顺序访问树中的元素。
解题关键是:在中序遍历过程中使用一个变量来记录当前节点被访问的次序。
代码如下:
// ac地址:https://leetcode-cn.com/problems/kth-smallest-element-in-a-bst/
// 原文地址:https://xxoo521.com/2020-03-05-kth-smallest-element-in-a-bst/
/**
* @param {TreeNode} root
* @param {number} k
* @return {number}
*/
var kthSmallest = function(root, k) {
let i = 0;
let val = null;
travel(root);
return val;
function travel(node) {
node.left && travel(node.left);
if (++i === k) {
val = node.val;
return;
}
node.right && travel(node.right);
}
};