二叉搜索树的第k个结点(简单,树)

题目描述
给定一棵二叉搜索树,请找出其中的第k小的TreeNode结点。
示例1
输入
{5,3,7,2,4,6,8},3
返回值
{4}
说明
按结点数值大小顺序第三小结点的值为4
二叉搜索树的中序遍历既是从小到大排序好的数列

/*
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) {
    
    
        if(pRoot==NULL||k<=0) return NULL;
        stack<TreeNode* >t;
        while(pRoot!=NULL||t.size())
        {
    
    
            if(pRoot!=NULL)
            {
    
    
                t.push(pRoot);
                pRoot=pRoot->left;
            }
            else
            {
    
    
                pRoot=t.top();
                t.pop();
                k--;
                if(k==0) return pRoot;
                pRoot=pRoot->right;
            }
        }
        return NULL;
    }
};

猜你喜欢

转载自blog.csdn.net/weixin_43540515/article/details/114329479