- 二叉搜索树中第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
提示:
树中的节点数为 n 。
1 <= k <= n <= 104
0 <= Node.val <= 104
解法:中序遍历序列化所有节点指针或节点值,取第 kk 个即可。
优化:添加全局计数器,记录当前是中序遍历遍历到的第几个节点,如果为 kk ,则将该节点的信息保存在全局量中,提前结束遍历。
/**
* 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 cnt = 0;
int topk;
int kthSmallest(TreeNode* root, int k) {
inorder(root, k);
return topk;
}
void inorder(TreeNode* node, int k) {
if (!node) return;
inorder(node->left, k);
++cnt;
if (cnt == k) {
topk = node->val;
return;
}
inorder(node->right, k);
}
};
时间复杂度:平均、最坏 O(N)O(N)。遍历过程访问每个节点一次。
空间复杂度:栈空间 O(h)O(h),hh 为树高。平均 h = lgNh=lgN,最坏 h = Nh=N。
作者:lzx071021
链接:https://leetcode-cn.com/problems/kth-smallest-element-in-a-bst/solution/bst-zhong-de-topk-wen-ti-by-lzx071021-ky43/