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

本姑凉不耐代码

于 2020-07-05 16:47:47 发布

阅读量2.1k

点赞数

CC 4.0 BY-SA版权

文章标签：算法二叉排序树

本文链接：https://blog.youkuaiyun.com/mn08290125/article/details/107141010

本文介绍了一种在二叉搜索树中查找第k个最小元素的方法，通过中序遍历的方式，实现了kthSmallest函数。文章详细解释了正确的递归实现方式，并指出了常见错误写法及其原因，强调了返回值类型的重要性。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

leetcode刷题

给定一个二叉搜索树，编写一个函数 kthSmallest 来查找其中第 k 个最小的元素。
说明：
你可以假设 k 总是有效的，1 ≤ k ≤ 二叉搜索树元素个数。

class Solution {
    int count = 0;
    int res = -1;
    public int kthSmallest(TreeNode root, int k) {
        inOrder(root,k);
        return res;       
    }
    private void inOrder(TreeNode node,int k){
        if(node!=null){
            inOrder(node.left,k);
            count++;
            if(k==count){
                res = node.val;
                return;
            }
            inOrder(node.right,k);
        }
    }
}

错误写法：注意返回值是什么类型！！！

class Solution {
    public int kthSmallest(TreeNode node, int k) {
        if(node!=null){
            kthSmallest(node.left,k);//这样写为什么不对呢？是因为它的返回值就是int类型的了，而不是中序遍历了！！！1
            count++;
            if(k==count){
                res = node.val;
                return;
            }
            kthSmallest(node.right,k);
        }    
    }
}