思路:二叉搜索树的中序遍历是递增序列,将中序遍历的结果保存下来,返回中序遍历的倒数第k个元素即可。
二叉搜索树第 k 大的节点可转化为求 此树的中序遍历的倒数第 k 个节点
class Solution {
public int kthLargest(TreeNode root, int k) {
List<Integer> res = new ArrayList<>();
inorder(root, res);
// 返回中序遍历的倒数第k个元素
return res.get(res.size() - k);
}
public void inorder(TreeNode root, List<Integer> list){
if (root == null) return ;
// 左
inorder(root.left, list);
// 根
list.add(root.val);
// 右
inorder(root.right, list);
}
}
改进:中序遍历是 左->根->右
,如果我们右->根->左
遍历,即可得到中序遍历的逆序,同时我们可以不使用List保存遍历结果,而是找到了直接返回,空间复杂度从O(N)降低到O(1)
二叉搜索树第 k 大的节点可转化为求 此树的中序遍历的倒序的第 k 个节点
class Solution {
int ans = 0, count = 0;
public int kthLargest(TreeNode root, int k) {
inorder(root, k);
return ans;
}
void inorder(TreeNode root, int k){
if (root == null) return;
// 右
inorder(root.right, k);
// 根
if (++count == k){
ans = root.val;
// 提前结束
return;
}
// 左
inorder(root.left, k);
}
}
迭代版本:
class Solution {
public int kthLargest(TreeNode root, int k) {
Deque<TreeNode> stack = new LinkedList<>();
while (!stack.isEmpty() || root != null){
while(root != null){
stack.push(root);
root = root.right;
}
root = stack.peek();
stack.pop();
if (--k == 0) return root.val;
root = root.left;
}
return 0;
}
}