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

给定一个二叉搜索树的根节点 root ，和一个整数 k ，请你设计一个算法查找其中第 k 小的元素（从 1 开始计数）。

示例 1：

输入：root = [3,1,4,null,2], k = 1
输出：1

示例 2：

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

思路

        中序遍历

代码

        迭代

class Solution {
public:
    vector<int> res;
    inline void dfs(TreeNode* node)
    {
        if(node == nullptr)
            return;
        dfs(node->left);
        res.push_back(node->val);
        dfs(node->right);
    }
    int kthSmallest(TreeNode* root, int k) {
        TreeNode* node = root;
        dfs(node);
        return res[k-1];
    }
};

        递归

class Solution {
public:
    vector<int> res;
    inline void dfs(TreeNode* node)
    {
        if(node == nullptr)
            return;
        dfs(node->left);
        res.push_back(node->val);
        dfs(node->right);
    }
    int kthSmallest(TreeNode* root, int k) {
        TreeNode* node = root;
        dfs(node);
        return res[k-1];
    }
};