给定一棵二叉搜索树,请找出其中第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
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/er-cha-sou-suo-shu-de-di-kda-jie-dian-lcof
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
第一种方法,使用 ArrayList 存放中序遍历的结果,然后在数组中找到倒数第 k 个元素即可,但是这种方法时间和空间复杂度都很高,因为不管找第几个元素,都要中序遍历走完,下面的那个是效率高的方法:
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
List<Integer> list = new ArrayList<>();
public int kthLargest(TreeNode root, int k) {
midSearch(root);
return list.get(list.size() - k);
}
public void midSearch(TreeNode node) {
if (node != null) {
midSearch(node.left);
list.add(node.val);
midSearch(node.right);
}
}
}
力扣崩了…一会儿再写
第二种方法,空间复杂度低,时间复杂度低,只需要将二叉树的中序遍历反过来,变成 右根左, 即可,这样倒数第 k 项大的就变成了中序遍历第 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 {
int k, res;
public int kthLargest(TreeNode root, int k) {
this.k = k;
midSearch(root);
return res;
}
public void midSearch(TreeNode node) {
if (node != null) {
midSearch(node.right);
k--;
if (k == 0) {
res = node.val;
return;
}
midSearch(node.left);
}
}
}