剑指 Offer 54. 二叉搜索树的第k大节点
难度:简单
给定一棵二叉搜索树,请找出其中第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 ≤ 二叉搜索树元素个数
解答:
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
//DFS
//时间复杂度:O(N)。空间复杂度:O(N)
int ans, k;
public int kthLargest(TreeNode root, int k) {
this.k = k;
DFS(root);
return ans;
}
void DFS(TreeNode root) {
if(root == null || k == 0) return;
DFS(root.right);
if(k == 0) return;
if(--k == 0){
ans = root.val;
return;
}
DFS(root.left);
}
}
参考自:
作者:LeetCode-Solution
链接: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)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。