给定一棵二叉搜索树,请找出其中第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 ≤ 二叉搜索树元素个数3
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
//这道题大神做法太牛逼了,中序遍历的倒序遍历[右子树|根节点|左子树],在定义一个k,
//就减少了空间复杂度,当k减小到0时,就把值床给res,结束循环
int res, k;
public int kthLargest(TreeNode root, int k) {
this.k = k;
dfs(root);
return res;
}
void dfs(TreeNode node){
if(node == null) return;
dfs(node.right);
if(k-- == 1){//注意这里k-- 和--k有很大区别
res = node.val;
return;
}
dfs(node.left);
}
}