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

Lambda_eta

于 2019-06-01 09:40:05 发布

阅读量137

点赞数 1

CC 4.0 BY-SA版权

分类专栏： LeetCode/C++

本文链接：https://blog.youkuaiyun.com/Zolewit/article/details/90715708

LeetCode/C++ 专栏收录该内容

248 篇文章

订阅专栏

本文介绍了一种使用栈实现的中序遍历算法，该算法能够在二叉搜索树中找到第K小的元素。通过迭代而非递归的方式，此方法有效地避免了深度递归可能带来的栈溢出问题。

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

在这里插入图片描述
中序遍历

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    int kthSmallest(TreeNode* root, int k) {
        stack<TreeNode*> stk;
        int count=0;
        TreeNode* cur=root;
        while(cur || !stk.empty()){
            while(cur){
                stk.push(cur);
                cur=cur->left;
            }
            cur=stk.top();
            stk.pop();
            
            if(++count==k)
                break;
            
            cur=cur->right;
        }
        return cur->val;
    }
};