题目描述
给定一棵二叉搜索树,请找出其中的第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;
}
};