给定一棵二叉搜索树,请找出其中第 k 大的节点的值。
简单难度,我们使用反中序遍历,然后遍历一次进行k自减1,当k为0时对应的元素就是第k大的节点。
int k, res;
public int kthLargest(TreeNode root, int k) {
this.k = k;
dfs(root);
return res;
}
private void dfs(TreeNode root) {
//右递归
if (root.right != null && k > 0) {
dfs(root.right);
}
//中序递减k并判断是否为0
if (--k == 0) {
res = root.val;
return;
}
//左循环
if (root.left != null && k > 0) {
dfs(root.left);
}
}