找出二叉搜索树中第k小的元素(Java)

原创于 2019-03-16 20:35:35 发布 · 1.2k 阅读

3 ·

CC 4.0 BY-SA版权

树题专栏收录该内容

6 篇文章

订阅专栏

本文介绍了一种高效算法，用于在二叉搜索树中查找第k小的元素。通过迭代遍历左子树并使用栈保存节点，算法能在O(h)的空间复杂度下实现，其中h为树的高度。

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

题目描述

给出一个二叉搜索树，返回这个二叉搜索树中第k小的元素

解决

代码：

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