BST中第K小的元素

给一棵二叉搜索树，写一个 KthSmallest 函数来找到其中第 K 小的元素。

样例

样例 1:

输入：{1,#,2},2
输出：2
解释：
	1
	 \
	  2
第二小的元素是2。

样例 2:

输入：{2,1,3},1
输出：1
解释：
  2
 / \
1   3
第一小的元素是1。

挑战

如果这棵 BST 经常会被修改(插入/删除操作)并且你需要很快速的找到第 K 小的元素呢？你会如何优化这个 KthSmallest 函数？

注意事项

你可以假设 k 总是有效的， 1 ≤ k ≤ 树的总节点数。

 

/**
 * Definition of TreeNode:
 * class TreeNode {
 * public:
 *     int val;
 *     TreeNode *left, *right;
 *     TreeNode(int val) {
 *         this->val = val;
 *         this->left = this->right = NULL;
 *     }
 * }
 */

class Solution {
public:
    /**
     * @param root: the given BST
     * @param k: the given k
     * @return: the kth smallest element in BST
     */
     
    vector<int> nums;
    int kthSmallest(TreeNode * root, int k) {
        // write your code here
        bianli(root);
        sort(nums.begin(), nums.end());
        return nums[k-1];
    }
    void bianli(TreeNode * root)
    {
        if(root != NULL)
        {
            nums.push_back(root->val);
            bianli(root->left);
            bianli(root->right);
        }
    }
};