题意:给定一棵二叉搜索树,请找出其中的第k小的结点。例如, (5,3,7,2,4,6,8) 中,按结点数值大小顺序第三小结点的值为4。
思路:中序遍历就是从小到大排序,直接中序遍历一下就好了。
class Solution {
public:
TreeNode* KthNode(TreeNode* pRoot, int k)
{
if (pRoot != NULL)
{
TreeNode* node = KthNode(pRoot->left, k);
if (node) return node;
if (++count == k)
return pRoot;
node = KthNode(pRoot->left, k);
if (node) return node;
}
return NULL;
}
private:
int count = 0;
};