LeetCode 230 二叉搜索树中第k小元素

• 分析
  中序遍历得到递增数组，取第k个元素。时间复杂度O(N)，空间复杂度O(N)，其中N为数组的个数。可以使复杂度达到O（K），这就需要把递归函数优化一下，把K传进去，访问到第K个最小元素的时候返回。
• 代码

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    int kthSmallest(TreeNode* root, int k) {
        vector<int> vec;
        inorder(root, vec);
        return *(vec.begin() + k - 1);

    }

    void inorder(TreeNode* root, vector<int>& vec){
        if(root == nullptr) return;

        inorder(root -> left, vec);
        vec.push_back(root -> val);
        inorder(root -> right, vec);
    }
};
//O（K）代码
class Solution {
private:
    int res;    
public:
    int kthSmallest(TreeNode* root, int k) {
        inorder(root, k);
        return res;

    }

    void inorder(TreeNode* root, int& k){
        if(root == nullptr) return;

        inorder(root -> left, k);
        if(--k == 0){
            res = root -> val;
            return;
        }
        inorder(root -> right, k);
    }
};