剑指 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 ≤ 二叉搜索树元素个数
二、暴力法
因此本题提供的树是二叉搜索树,所以对这棵树进行中序遍历后的数组即为有序数组,再返回其第k大节点即可。
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
private List<Integer> list = new ArrayList<>();
public int kthLargest(TreeNode root, int k) {
inOrder(root);
return list.get(list.size() - k);
}
public void inOrder(TreeNode root){
if(root.left != null){
inOrder(root.left);
}
list.add(root.val);
if(root.right != null){
inOrder(root.right);
}
}
}
三、逆中序遍历法
因为二叉搜索树的中序遍历是递增序列。
所以本题可以转化为求此二叉搜索树的倒序中序遍历的第k个节点。
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
int k, num;
public int kthLargest(TreeNode root, int k) {
this.k = k;
reverseInOrder(root);
return num;
}
public void reverseInOrder(TreeNode root){
if(root.right != null){
reverseInOrder(root.right);
}
k--;
if(k == 0){
//若 k = 0 ,代表已找到目标节点,无需继续遍历,赋值后直接返回
num = root.val;
return;
}
if(root.left != null){
reverseInOrder(root.left);
}
}
}