剑指二叉搜索树的第K个结点

twh233

于 2019-09-02 17:50:21 发布

阅读量139

点赞数

分类专栏：剑指

本文链接：https://blog.youkuaiyun.com/qq_36553623/article/details/100335733

版权

剑指专栏收录该内容

10 篇文章

订阅专栏

本文介绍了一种使用栈实现的非递归方法，来找到二叉树中序遍历的第K个节点。通过迭代地将左子节点压入栈中，然后弹出并计数，直到计数等于K，此时栈顶元素即为所求节点。

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

题目链接

非递归做法：

/*
struct TreeNode {
    int val;
    struct TreeNode *left;
    struct TreeNode *right;
    TreeNode(int x) :
            val(x), left(NULL), right(NULL) {
    }
};
*/
class Solution {
public:
    TreeNode* KthNode(TreeNode* pRoot, int k)
    {
        TreeNode* tmp = pRoot;
        if(pRoot == NULL) return NULL;
        stack<TreeNode*> q;
        while(tmp != NULL || !q.empty()) {
            if(tmp != NULL) {
                q.push(tmp);
                tmp = tmp->left;
            }
            else {
                TreeNode* top = q.top();
                q.pop();
                k--;
                if(k == 0) return top;
                tmp = top->right;
            }
        }
        return tmp; 

    }

    
};