【剑指】72,二叉搜索树的第k个节点

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

你可能感兴趣的:(剑指offer)