剑指 Offer 54. 二叉搜索树的第k大节点
给定一棵二叉搜索树,请找出其中第k大的节点。
输入: root = [3,1,4,null,2], k = 1
3
/ \
1 4
\
2
输出: 4
输入: root = [5,3,6,2,4,null,null,1], k = 3
5
/ \
3 6
/ \
2 4
/
1
输出: 4
这题应该也是中序遍历就OK了,但是这里由于是要拿第K大的元素,所以遍历顺序应该为右->中->左
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);
}
}