面试题54：二叉搜索树的第k个结点

题目描述
给定一棵二叉搜索树，请找出其中的第k小的结点。例如， （5，3，7，2，4，6，8） 中，按结点数值大小顺序第三小结点的值为4。

/*
struct TreeNode {
    int val;
    struct TreeNode *left;
    struct TreeNode *right;
    TreeNode(int x) :
            val(x), left(NULL), right(NULL) {
    }
};
*/
class Solution {
public:
    TreeNode* inOrderHelp(TreeNode* pRoot, int& k)
    {
        TreeNode* aim = nullptr;
        if(pRoot->left != nullptr)
            aim = inOrderHelp(pRoot->left, k);
        if(aim == nullptr)//递归出口
        {
            if(k == 1)
                aim = pRoot;//k==1,即为当前节点
            k --;
        }
        if(aim == nullptr && pRoot->right != nullptr)
            aim = inOrderHelp(pRoot->right, k);
        return aim;
    }

    TreeNode* KthNode(TreeNode* pRoot, int k)
    {
        if(pRoot == nullptr || k == 0)
            return nullptr;
        return inOrderHelp(pRoot, k);
    }
};