求二叉搜索树的第k个节点

本题源自剑指offer

-----------------------------------------------------------------------------

用中序遍历二叉搜索树得到有序数列，即可得到第k个数

递归：

TreeNode* KthNode(TreeNode* pRoot, int k)
{
if(pRoot==NULL || k==0)
return NULL;
return KthNodeCore(pRoot,k);
}
TreeNode* KthNodeCore(TreeNode* root,int& k){
TreeNode* target=NULL;
if(root->left!=NULL){
target=KthNodeCore(root->left,k);
}
if(target==NULL){
if(k==1)
target=root;
k--;
}
if(target==NULL&&root->right!=NULL){
target=KthNodeCore(root->right,k);
}
return target;
}

非递归：

TreeNode* KthNode(TreeNode* pRoot, int k)
{
if(pRoot==NULL || k==0)
return NULL;
stack<TreeNode*> treeStack;
TreeNode* result=NULL;
int count=0;
TreeNode* p=pRoot;
while(p!=NULL || !treeStack.empty()){
while(p!=NULL){
treeStack.push(p);
p=p->left;
}
TreeNode* q=treeStack.top();
treeStack.pop();
count++;
if(count==k){
result=q;
}
p=q->right;
}
return result;
}