目录
题解
本文解法基于此性质:二叉搜索树的中序遍历为 递增序列
根据以上性质,易得二叉搜索树的 中序遍历倒序 为 递减序列 。
因此,求 “二叉搜索树第 k 大的节点” 可转化为求 “此树的中序遍历倒序的第 k 个节点”。
题目指出:1≤k≤N (二叉搜索树节点个数);因此无需考虑 k > N 的情况。
若考虑,可以在中序遍历完成后判断 k>0 是否成立,若成立则说明 k > N 。
代码
class Solution {
int res, k;
public int kthLargest(TreeNode root, int k) {
this.k = k;
dfs(root);
return res;
}
void dfs(TreeNode root) {
if(root == null) return;
dfs(root.right);
if(k == 0) return;
if(--k == 0) res = root.val;
dfs(root.left);
}
}