【剑指Offer】二叉搜索树的第k个结点

1.题目

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

2.Code

二叉搜索树的中序遍历就是有序序列，定义一个计数器count每次中序遍历的时候+1，当等于k的时候结束

/*
struct TreeNode {
    int val;
    struct TreeNode *left;
    struct TreeNode *right;
    TreeNode(int x) :
            val(x), left(NULL), right(NULL) {
    }
};
*/
class Solution {
private:
    TreeNode* res;
    int count; 
public:
    Solution(){
        res = NULL;
        count = 0;
    }
    TreeNode* KthNode(TreeNode* pRoot, int k)
    {
        if(pRoot == NULL || k <= 0) return NULL;
        FindKth(pRoot,k);
        return res;
    }

    void FindKth(TreeNode* tree, int k)
    {
        if(tree == NULL)   return ;
        if(tree->left)
            FindKth(tree->left,k);
        count++;
        if(count == k)
            res = tree;
        if(tree->right)
            FindKth(tree->right,k);
    }
    
};