二叉搜索树的第k大节点
给定一棵二叉搜索树,请找出其中第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 ≤ 二叉搜索树元素个数
题解:中文力扣中的题意和书上的不太一致,书上是输出从小到大的第k个数,力扣上输出从大到小的第k个数。
所以在力扣上的话要将中序遍历的顺序改一下,改为右-中-左 的顺序进行遍历。当到达第k个数的时候就是目标值。
/**
* 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 {
private:
int result = -1;
public:
int kthLargest(TreeNode* root, int k) {
if(root == nullptr || k < 0)
return -1;
inOrder(root,k);
return result;
}
void inOrder(TreeNode* root, int& k){
if(root == nullptr)
return;
if(root->right != nullptr)
inOrder(root->right,k);
if(k == 1){
result = root->val;
k = k - 1;
return;
}
k = k-1;
if(root->left != nullptr)
inOrder(root->left,k);
}
};