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

知识搬运工er

于 2022-07-22 08:48:06 发布

阅读量341

点赞数

CC 4.0 BY-SA版权

分类专栏： Leetcode题解文章标签： leetcode 算法数据结构

本文链接：https://blog.youkuaiyun.com/hljjaaa/article/details/125924691

Leetcode题解专栏收录该内容

47 篇文章

订阅专栏

这篇博客介绍了如何在二叉搜索树中找到第K小的元素，提供了两种方法：一是将树转换为有序数组再查找，二是使用迭代的中序遍历。这两种方法都涉及到树的遍历和排序概念，适合数据结构和算法的学习者参考。

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

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

在这里插入图片描述

方法一、把二叉搜索树转换成有序数组（递归）

/**
    1、转换成有序数组
 */
public int kthSmallest(TreeNode root, int k) {
    List<Integer> list = new ArrayList<>();
    inOrder(root, list);
    return list.get(k - 1);
}

public void inOrder(TreeNode root, List<Integer> list) {
    if(root == null) return;
    inOrder(root.left, list);
    list.add(root.val);
    inOrder(root.right, list);
}

方法二、迭代(中序遍历的迭代法）

迭代的时候，每次弹出一个节点的时候，k–,当k = 0的时候，说明找到了第k小的节点，返回节点对应的值

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