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

最新推荐文章于 2025-04-29 08:00:00 发布

原创最新推荐文章于 2025-04-29 08:00:00 发布 · 142 阅读

0 ·

CC 4.0 BY-SA版权

二叉树专栏收录该内容

20 篇文章

订阅专栏

本文介绍了一种在二叉搜索树中查找第K个最小元素的方法，通过中序遍历的特性，实现了高效的查找算法。示例展示了具体的输入输出情况，帮助理解算法的运行过程。

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

描述：

给定一个二叉搜索树，编写一个函数 kthSmallest 来查找其中第 k 个最小的元素。

说明：
你可以假设 k 总是有效的，1 ≤ k ≤ 二叉搜索树元素个数。

示例 1:

输入: root = [3,1,4,null,2], k = 1
   3
  / \
 1   4
  \
   2
输出: 1

示例 2:

输入: root = [5,3,6,2,4,null,null,1], k = 3
       5
      / \
     3   6
    / \
   2   4
  /
 1
输出: 3

思路分析：

1、参考中序遍历输出是有序的特性

代码实现：

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    
    private int idx = 1;
    private int res;
    
    private void idOrder(TreeNode root,int k){
        if (root == null) return;
        idOrder(root.left, k);
        if(idx > k) return ; //相当于设置了一个返回的标记，表示已经找到第k小的值了，不用继续向下寻找了。
        if(idx == k){
            res = root.val;
        } 
        idx++;

        idOrder(root.right, k);
    }
    
    public int kthSmallest(TreeNode root, int k) {
        idOrder(root,k);
        return res;
    }
}