给定一棵二叉搜索树,请找出其中第k大的节点。
示例 1:
输入: root = [3,1,4,null,2], k = 1
3
/ \
1 4
\
2
输出: 4
示例 2:
输入: root = [5,3,6,2,4,null,null,1], k = 3
5
/ \
3 6
/ \
2 4
/
1
输出: 4
限制:
1 ≤ k ≤ 二叉搜索树元素个数
题目链接:https://leetcode-cn.com/problems/er-cha-sou-suo-shu-de-di-kda-jie-dian-lcof/submissions/
思路
法一:逆中序遍历
BST的中序便利是递增序列,因此逆中序遍历到第k个即为所求。
时间复杂度O(n),空间复杂度O(n)。
/**
* 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:
void dfs(TreeNode* root, int &k, int &res){
if(res>INT_MIN || !root){
return;
}
dfs(root->right, k, res);
--k;
if(k==0){
res = root->val;
return;
}
dfs(root->left, k, res);
}
int kthLargest(TreeNode* root, int k) {
int res = INT_MIN;
dfs(root, k, res);
return res;
}
};
法二:左右子树节点数
第k个节点的右子树节点都更大,因此当右子树节点数为k-1时,当前节点为所求。
/**
* 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:
int cnt(TreeNode* root){
if(!root) return 0;
return cnt(root->left) + cnt(root->right) + 1;
}
int kthLargest(TreeNode* root, int k) {
int rchild = cnt(root->right);
if(rchild==k-1){
return root->val;
}else if(rchild>k-1){
return kthLargest(root->right, k);
}else{
return kthLargest(root->left, k-rchild-1);
}
}
};