题目
给定一棵二叉搜索树,请找出其中第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 ≤ 二叉搜索树元素个数
思路
对于二叉搜索树,有这样一个性质:对其进行中序遍历时,结果为结点的增序排列。中序遍历即按照左根右进行遍历。因此可以知道,如果按照右根左进行遍历,那么就是节点的减序排列
题解
C++
代码思路:
1.参考"标准形式"的dfs代码,仍将遍历函数inorder设计成void类型,通过全局变量data
记录结果,可以省去传参的麻烦。在递归程序中,可以注意这种设计。对于一个类对象,其成员变量即是一个"全局变量"
。
2. 程序中我将data初始化为了INT_MIN
,这是climits.h
中宏定义的量,其值为负二十一亿多(懂的都懂(bushi
class Solution {
public:
int count = 0;
int data = INT_MIN;
void inorder(TreeNode* root, int k) {
if (root == nullptr) return;
inorder(root->right, k);
if (++count == k)
data = root->val;
inorder(root->left, k);
}
int kthLargest(TreeNode* root, int k) {
inorder(root, k);
return data;
}
};
java
class Solution {
int count = 0;
int data = 0;
void inorder(TreeNode root, int k){
if (root == null) return;
inorder(root.right, k);
if(++count == k)
data = root.val; //注意访问权限,外包中的类,必须将其声明为public,这里才能访问
inorder(root.left, k);
}
public int kthLargest(TreeNode root, int k) {
inorder(root, k);
return data;
}
}