leetcode 二叉搜索树中第K小的元素

给定一个二叉搜索树，编写一个函数 kthSmallest 来查找其中第 k 个最小的元素。

说明：
你可以假设 k 总是有效的，1 ≤ k ≤ 二叉搜索树元素个数。

示例 1:

输入: root = [3,1,4,null,2], k = 1
3
/
1 4

2
输出: 1
示例 2:

输入: root = [5,3,6,2,4,null,null,1], k = 3
5
/
3 6
/
2 4
/
1
输出: 3

用中序遍历把结点的值存进数组里面，则数组是按照从小到大排列的，直接就可以返回答案

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    int kthSmallest(TreeNode* root, int k) {
        if(root == NULL) {
            return 0;
        }
        f(root);
        if(num.size() < k) {
            return 0;
        }
        return num[k-1];
    }
    vector<int> num;
    void f(TreeNode * root) {
        if(root != NULL && root->left != NULL) {
            f(root->left);
        }
        num.push_back(root->val);
        //std::cout<<num<<" ";
        if(root != NULL && root->right != NULL) {
            f(root->right);
        }
    }
};