www 上海网站建设网络seo优化公司

给定一个二叉搜索树的根节点 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];}
};