题目描述
给定一棵二叉搜索树,请找出其中的第k小的结点。例如, (5,3,7,2,4,6,8) 中,按结点数值大小顺序第三小结点的值为4
思路:
就是一个二叉搜索树遍历的过程,然后从最左节点开始,左中右(中序遍历)定义一个类全局变量count来进行–;ans作为值-结果类型参数传入
代码:
/*
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
TreeNode(int x) :
val(x), left(NULL), right(NULL) {
}
};
*/
//中序遍历,找到第k个。注意是第三小节点。
class Solution {
private:
int count;
public:
TreeNode* KthNode(TreeNode* pRoot, int k)
{
count=k;
if(count==0 ||pRoot==nullptr)
return nullptr;
TreeNode* ans=nullptr;
//值-结果
inorder(pRoot,ans);
return ans;
}
void inorder(TreeNode* p,TreeNode* &ans)
{
if(!p)
{
inorder(p->left,ans);
count--;
if(!count)
{
ans=p;
]
inorder(p->right,ans);
}
return ;
}