题目
给定一棵二叉搜索树,请找出其中第k大的节点。
思路
中序遍历二叉搜索树,是会按照从大到小(右中左)或从小到大(左中右)的顺序打印出二叉搜索树中的值。
那我们就在第k次将结果保留起来就好了,当过了第k次以后,后面的遍历就不需要了。
易错点:k在递归的过程中是全局变量,不能写在递归函数里,因为k是针对整个函数,只进行k次
图片
作者:jyd
链接:https://leetcode-cn.com/problems/er-cha-sou-suo-shu-de-di-kda-jie-dian-lcof/solution/mian-shi-ti-54-er-cha-sou-suo-shu-de-di-k-da-jie-d/
来源:力扣(LeetCode)
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);
}
}
遍历
class Solution {
int res, k;
public int kthLargest(TreeNode root, int k) {
if(root == null || k < 0) return -1;
this.k = k;
dfs(root);
return res;
}
private void dfs(TreeNode root) {
Stack<TreeNode> stack = new Stack<TreeNode>();
TreeNode cur = root;
while(!stack.isEmpty() || cur != null) {
while(cur != null) {
stack.push(cur);
cur = cur.right;
}
if(k == 0) return;
TreeNode node = stack.pop();
k--;
if(k == 0) res = node.val;
if(node.left != null) {
cur = node.left;
}
}
}
}