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

最新推荐文章于 2024-01-29 08:15:00 发布

qq_1213701132

最新推荐文章于 2024-01-29 08:15:00 发布

阅读量88

点赞数

分类专栏： leetcode

本文链接：https://blog.csdn.net/qq_43604517/article/details/103943317

版权

leetcode 专栏收录该内容

195 篇文章 0 订阅

订阅专栏

在这里插入图片描述
方法一：递归-中序遍历

class Solution {
private:
    vector<int> vc; //保存中序遍历结果
    void inorder(TreeNode* root){
        if(!root)
            return;
        inorder(root->left);
        vc.push_back(root->val);
        inorder(root->right);
    }
public:
    int kthSmallest(TreeNode* root, int k) {
        inorder(root);
        return vc[k-1];
    }
};

方法二：递归-中序遍历，提前结束

class Solution {
private:
    int cnt;
    int res;
    void inorder(TreeNode* root){
        if(!root || !cnt)   //遍历访问结点数达到返回
            return;
        inorder(root->left);
        if(--cnt==0)
            res=root->val;
        inorder(root->right);
    }
public:
    int kthSmallest(TreeNode* root, int k) {
        cnt=k;
        inorder(root);
        return res;
    }
};

方法三：迭代法-中序遍历，提前结束

class Solution {
public:
    int kthSmallest(TreeNode* root, int k) {
        stack<TreeNode*> st;
        int cnt=0;
        while(!st.empty() || root){
            while(root){
                st.push(root);
                root=root->left;
            }
            TreeNode* top=st.top();
            st.pop();
            if(++cnt==k)
                return top->val;
            root=top->right;
        }
        return -1;
    }
};