题目描述:
解法思路:
根据题目意思,要找二叉搜索树的第k大节点。那么我们可以想一下,如果我们要找二叉树的最小节点要怎么找呢?直接中序遍历就能找到最左的节点,最左的节点也就是最小的节点。同理,我们要找最大节点,只需要把中序遍历改一下即可,中序遍历的顺序是左 -> 根 -> 右,那么我们改成右 -> 根 -> 左。因为题目是找到第k大,那么直接判断k是否等于1,k不等于1,就k减去1,返回上一层,当k等于1,记录节点值。最后返回节点值即可。
代码实现:
class Solution {
int count; int k;
public int kthLargest(TreeNode root, int k) {
this.k = k;
dfs(root);
return this.count;
}
void dfs(TreeNode root){
if(root == null) return ;
dfs(root.right);
if(k == 1){
this.count = root.val; k--; return ;
}
k--;
dfs(root.left);
}
}
执行结果: