62.二叉搜索树的第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* KthNode(TreeNode* pRoot, int k)
    {
        if(pRoot == nullptr || k == 0)
            return nullptr;
        vector<TreeNode*> v;
        KthNode(pRoot,v);
        if(k <= v.size())
            return v[k - 1];
        return nullptr;
    }
    void KthNode(TreeNode* pRoot, vector<TreeNode*>& v)
    {
        if(pRoot == nullptr)
            return;
        if(pRoot->left)
           KthNode(pRoot->left,v);
        v.push_back(pRoot);
        if(pRoot->right)
            KthNode(pRoot->right,v);
    }
};