【重点】【二叉树】230.二叉搜索树中第K小的元素

原创已于 2025-03-21 01:25:19 修改

· 400 阅读

9 ·

版权

文章标签：

#二叉树

于 2023-12-06 22:39:48 首次发布

力扣Top100 同时被 2 个专栏收录

100 篇文章

订阅专栏

labuladong的算法小抄笔记

78 篇文章

订阅专栏

本文介绍了一种使用Java编写的中序遍历算法，用于在二叉树中找到第k小的元素。通过递归方法，先遍历左子树，将节点值添加到结果列表，然后遍历右子树，最后返回列表中第k个元素。

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

题目

法1：中序遍历

Python

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    def kthSmallest(self, root: Optional[TreeNode], k: int) -> int:
        res = list()
        self.mid_travel(root, res, k)

        return res[k-1]

    def mid_travel(self, root: Optional[TreeNode], res: List[int], k):
        if root == None:
            return 
        self.mid_travel(root.left, res, k)
        res.append(root.val)
        if len(res) == k:
            return 
        self.mid_travel(root.right, res, k)

Java

最简练的写法，牢记！！！

class Solution {
    public int kthSmallest(TreeNode root, int k) {
        List<Integer> res = new ArrayList<>();
        find(root, res, k);

        int lastIndex = Math.min(res.size() - 1, k - 1);
        return res.get(lastIndex);
    }

    public void find(TreeNode root, List<Integer> res, int k) {
        if (root == null || res.size() == k) {
            return;
        } 
        find(root.left, res, k);
        res.add(root.val);
        find(root.right, res, k);
    }
}