给定一棵二叉搜索树,请找出其中第 k 大的节点的值。
示例 1:
输入: root = [3,1,4,null,2], k = 1
3
/ \
1 4
\
2
输出: 4
示例 2:
输入: root = [5,3,6,2,4,null,null,1], k = 3
5
/ \
3 6
/ \
2 4
/
1
输出: 4
限制:
1 ≤ k ≤ 二叉搜索树元素个数
解题思路:,搜索二叉树右子树大于根节点,左子树小于根节点。按照右根左的顺序递归遍历,同时k--,判断k=1时的值就是要返回的值。
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
Stack<Integer> stack = new Stack<Integer>();
int ret, k;
public int kthLargest(TreeNode root, int k) {
this.k = k;
dfs(root);
return ret;
}
public void dfs(TreeNode root){
if(root == null || k == 0){
return;
}
dfs(root.right);
if(--k == 0){
ret = root.val;
return;
}
dfs(root.left);
}
}
注意:这里的k的用法,全局变量。
记一种错误写法,k作为形参传递:
class Solution {
Stack<Integer> stack = new Stack<Integer>();
int ret;
public int kthLargest(TreeNode root, int k) {
dfs(root, k);
return ret;
}
public void dfs(TreeNode root, int k){
if(root == null || k == 0){
return;
}
dfs(root.right, k);
if(--k == 0){
ret = root.val;
return;
}
dfs(root.left, k);
}
}
这种方法在回溯时k值是会变回去的,不是--k后的值。