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

关注

原创最新推荐文章于 2026-04-23 10:46:17 发布 198 阅读 AI 写同款· GEO 优化 ›

Python学习（一）快速入门 1.Linux系统中是可以直接使用Python进行编程，具体如下：进入Python：在linux的Terminal中输入 Python ，进入Python的命令行；退出Python：在linux的Terminal中输入quit() 或 exit() 或 ctrl +d ，再回车，退出Python的命令行。 2.python的源程序就是一个特殊格式的文本文件，可以使用任意文本编辑软件做Python的开发。 3.Python程序的文件扩展名通常都是 .py。 4.执行Python程序的方式：在阅读详情

1、题目

在这里插入图片描述

2、解法1（递归法）

1、错误的解法（递归使用的深度遍历DFS,不可能在–k==0处停止，所以以下这种方式不合理）
在这里插入图片描述
2、正确的递归

class Solution {
    public ArrayList<Integer> inorder(TreeNode t, ArrayList<Integer> arr) {
        if (t == null) return arr;
        inorder(t.left, arr);
        arr.add(t.val);
        inorder(t.right, arr);
        // arr必须是从底层返回，所以身为管理者
        // 必须一层层传递下去
        return arr;
    }
    public int kthSmallest(TreeNode root, int k) {
        // 可以由相同的元素赋值
        ArrayList<Integer> arr = inorder(root, new ArrayList<Integer>());
        return arr.get(k-1);
    }
}

时间复杂度为O(N),空间复杂度为O(N)

解法2：使用栈(利用了二叉搜索树的中序遍历就是升序数列）

class Solution {
    public int inorder(TreeNode t, int k) {
        LinkedList<TreeNode> stk = new LinkedList<>();
        while (!stk.isEmpty() || t != null) {
            while (t != null) {
                stk.add(t);
                t = t.left;
            }
            t = stk.removeLast();
            if (--k == 0) return t.val;
            t = t.right;
            }
        return -1;
    }
    public int kthSmallest(TreeNode root, int k) {
        return inorder(root, k);
    }
}