题目描述
给定一棵二叉搜索树,请找出其中的第k小的结点。例如, (5,3,7,2,4,6,8) 中,按结点数值大小顺序第三小结点的值为4。
思路:中序遍历第k个节点
TreeNode* KthNode(TreeNode* pRoot, int k)
{
//中序遍历
int count=0;
TreeNode*p=pRoot;
stack<TreeNode*>s;
if(pRoot==NULL) return p;
do{
while(p!=NULL)
{
s.push(p);
p=p->left;
}
p=s.top();
s.pop();
count++;
if(count==k) return p;
p=p->right;
}while(p!=NULL||!s.empty());
return p;
}
{
//中序遍历
int count=0;
TreeNode*p=pRoot;
stack<TreeNode*>s;
if(pRoot==NULL) return p;
do{
while(p!=NULL)
{
s.push(p);
p=p->left;
}
p=s.top();
s.pop();
count++;
if(count==k) return p;
p=p->right;
}while(p!=NULL||!s.empty());
return p;
}