leetcode 230. Kth Smallest Element in a BST

最新推荐文章于 2025-12-03 17:03:32 发布

原创最新推荐文章于 2025-12-03 17:03:32 发布 · 75 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#算法

leetcode 专栏收录该内容

490 篇文章

订阅专栏

本文介绍了一种在二叉搜索树中寻找第K小元素的算法。通过中序遍历的方式，可以有效地找到目标元素。该方法利用了二叉搜索树的特性，确保遍历顺序为升序，从而在遍历过程中实时判断是否找到第K小元素。

Given a binary search tree, write a function kthSmallest to find the kth smallest element in it.

Note:
You may assume k is always valid, 1 ≤ k ≤ BST’s total elements.

Example 1:

Input: root = [3,1,4,null,2], k = 1

Output: 1

输出二叉树中第K大的节点值

思路：
考虑到中序遍历是按顺序输出的，所以中序遍历，并记录count，当k==count时返回，
为了及时终止遍历，返回boolean类型

//1ms
class Example {
    int count = 1;
    int res = 0;
    
    public int kthSmallest(TreeNode root, int k) {
        inorder(root, k);
        return res;
    }
    
    public boolean inorder(TreeNode root, int k) {
        if (root == null) {
            return false;
        }
        
        if(inorder(root.left, k)) {
            return true;
        };
        
        if (k == count) {
            res = root.val;
            return true;
        }
        
        count ++;
        
        if(inorder(root.right, k)) {
            return true;
        }
        
        return false;
    }
}