230. 二叉搜索树中第 K 小的元素
问题描述
给定一个二叉搜索树的根节点 root ,和一个整数 k ,请你设计一个算法查找其中第 k 小的元素(从 1 开始计数)。
分析
已经给定BST,找出第k大的数字。
- 可以利用BST性质:中序遍历一定是升序数组,遍历到第k个结点即为答案,复杂度O(n)
- 注意,java中成员变量名与函数形参重名,可以用this.x=x,C++中,可以用this->x=x解决
实现
class Solution {
public:
int x = 0;
int k;
int res;
void solve(TreeNode* root) {
if (root == nullptr) {
return ;
}
solve(root->left);
x++;
if (x == k) {
res = root->val;
}
solve(root->right);
return ;
}
int kthSmallest(TreeNode* root, int k) {
if (root == nullptr) {
return -1;
}
this->k = k;
solve(root);
return res;
}
};
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· 10年+ .NET Coder 心语 ── 封装的思维:从隐藏、稳定开始理解其本质意义
· 地球OL攻略 —— 某应届生求职总结
· 周边上新:园子的第一款马克杯温暖上架
· Open-Sora 2.0 重磅开源!
· 提示词工程——AI应用必不可少的技术