首页
学习
活动
专区
圈层
工具
发布

LeetCode 230.二叉搜索树中第K小的元素 - JavaScript

题目描述:给定一个二叉搜索树,编写一个函数 kthSmallest 来查找其中第 k 个最小的元素。

说明:你可以假设 k 总是有效的,1 ≤ k ≤ 二叉搜索树元素个数

解法:利用 BST 性质

根据二叉搜索树(BST)的性质:节点的权重大于左节点的权重,大于右节点的权重。因此,对 BST 进行中序遍历,就是按从小到大的顺序访问树中的元素。

解题关键是:在中序遍历过程中使用一个变量来记录当前节点被访问的次序。

代码如下:

代码语言:javascript
复制
// 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);
    }
};
下一篇
举报
领券