230. 二叉搜索树中第K小的元素
给定一个二叉搜索树的根节点 root ,和一个整数 k ,请你设计一个算法查找其中第 k 个最小元素(从 1 开始计数)。
示例 1:
输入:root = [3,1,4,null,2], k = 1
输出:1
> 代码
class Solution {
public:
vector<int> res;
int aaa;
void traversal(TreeNode* root,int k){
if(!root || res.size() >= k) return;
traversal(root->left,k);
res.push_back(root->val);
if(res.size() == k) aaa = root->val;
traversal(root->right,k);
}
int kthSmallest(TreeNode* root, int k) {
traversal(root,k);
return aaa;
}
};