二叉搜索树的第K个节点

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

一刷：

两种方法，递归和非递归用中序遍历。

问题主要出在写中序遍历的时候，一个是没有加判断条件k>0, 一个是没有考虑到k比实际取数组的值大1.

class Solution {
public:
    int count = 0;
    TreeNode* KthNode(TreeNode* pRoot, int k)
    {
        if(pRoot){
            TreeNode* res = KthNode(pRoot->left, k);
            if(res) return res;
            if(++count == k) return pRoot;
            res = KthNode(pRoot->right, k);
            if(res) return res;
        }
        return nullptr;
    }
 
     
};

*
struct TreeNode {
    int val;
    struct TreeNode *left;
    struct TreeNode *right;
    TreeNode(int x) :
            val(x), left(NULL), right(NULL) {
    }
};
*/
class Solution {
public:
    vector<TreeNode*> res;
    TreeNode* KthNode(TreeNode* pRoot, int k)
    {
        if(pRoot && k > 0){
            inOrder(res, pRoot);
            if(k <= res.size()) return res[k-1];
        }
        return nullptr;
    }
     
    void inOrder(vector<TreeNode*>& res, TreeNode* pRoot){
        if(pRoot == NULL) return;
        inOrder(res, pRoot->left);
        res.push_back(pRoot);
        inOrder(res, pRoot->right);
    }
};