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

最新推荐文章于 2025-04-12 14:28:05 发布

overdosed_

最新推荐文章于 2025-04-12 14:28:05 发布

阅读量80

点赞数

分类专栏： leetcode 文章标签： leetcode 算法 java

本文链接：https://blog.youkuaiyun.com/weixin_45904682/article/details/129869918

版权

leetcode 专栏收录该内容

82 篇文章

订阅专栏

文章介绍了两种方法来寻找二叉树中第k小的元素。第一种是使用广度优先搜索(BFS)的暴力解法，将树的所有节点值存储并排序，然后返回第k个元素。第二种方法利用了二叉搜索树的性质，通过中序遍历在遍历过程中找到第k小的节点值。

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

暴力解法:

class Solution {
    public int kthSmallest(TreeNode root, int k) {
        List<Integer> list = new ArrayList<>();
        Deque<TreeNode> queue = new LinkedList<>();
        queue.offer(root);
        while (!queue.isEmpty()){
            TreeNode temp = queue.poll();
            list.add(temp.val);
            if(temp.left != null){
                queue.offer(temp.left);
            }
            if(temp.right != null){
                queue.offer(temp.right);
            }
        }
        Integer[] arr = list.toArray(new Integer[list.size()]);
        Arrays.sort(arr);
        return arr[k - 1];
    }
}

利用二叉搜索树 + 中序遍历性质

class Solution {
    public int kthSmallest(TreeNode root, int k) {
        Deque<TreeNode> stack = new ArrayDeque<>();
        while (root != null || !stack.isEmpty()){
            while (root !=null){
                stack.push(root);
                root = root.left;
            }
            root = stack.pop();
            k--;
            if(k == 0){
                return root.val;
            }
            root = root.right;
        }
        return -1;
    }
}