题目
面试题54. 二叉搜索树的第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 ≤ 二叉搜索树元素个数
思路
二叉搜索树的中序遍历就是按照从大到小排序的,所以用中序遍历就好了。
代码
class Solution {
int k;
int res = 0;
public int kthLargest(TreeNode root, int k) {
this.k = k;
help(root);
return res;
}
public void help(TreeNode root) {
if (root.right != null)
help(root.right);
k = k - 1;
if (k == 0) {
res = root.val;
return;
}
if (root.left != null)
help(root.left);
}
}