给定一棵二叉搜索树,请找出其中的第 k小的结点。
你可以假设树和 k都存在,并且 1≤k≤树的总结点数。
样例
输入:root = [2, 1, 3, null, null, null, null] ,k = 3
2
/ \
1 3
输出:3
分析:
采用递归依次遍历左子树,根和右子树
每当访问根节点就让, 当k=0时,此时的根节点就是我们求得答案。
时间复杂度
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
TreeNode* res = new TreeNode(-1);
TreeNode* kthNode(TreeNode* root, int k) {
dfs(root, k);
return res;
}
void dfs(TreeNode* root, int& k) {
if(!root) return;
dfs(root->left, k);
if(!--k) res = root;
dfs(root->right, k);
}
};
网友评论