这篇文章主要介绍leetCode如何找出二叉搜索树的第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 ≤ 二叉搜索树元素个数
二叉搜索树的中序遍历就是元素递增的,根据中序遍历得到的数据即可解决。
import java.util.ArrayList;
import java.util.Comparator;
import java.util.List;
public class KthLargestTest3 {
public static void main(String[] args) {
TreeNode t1=new TreeNode(3);
TreeNode t2=new TreeNode(1);
TreeNode t3=new TreeNode(4);
TreeNode t4=new TreeNode(2);
t1.left=t2;
t1.right=t3;
t2.right=t4;
int k=1;
int kthLargest = kthLargest(t1, k);
System.out.println("kthLargest = " + kthLargest);
}
public static int kthLargest(TreeNode root, int k) {
if (root == null) {
return 0;
}
List<Integer> list = new ArrayList<>();
dfs(root, list);
list.sort(Comparator.reverseOrder());
return list.get(k - 1);
}
private static void dfs(TreeNode root, List<Integer> list) {
if (root == null) {
return;
}
if (root.left != null) {
dfs(root.left, list);
}
list.add(root.val);
if (root.right != null) {
dfs(root.right, list);
}
}
}
以上是“leetCode如何找出二叉搜索树的第k大节点”这篇文章的所有内容,感谢各位的阅读!希望分享的内容对大家有帮助,更多相关知识,欢迎关注亿速云行业资讯频道!
免责声明:本站发布的内容(图片、视频和文字)以原创、转载和分享为主,文章观点不代表本网站立场,如果涉及侵权请联系站长邮箱:is@yisu.com进行举报,并提供相关证据,一经查实,将立刻删除涉嫌侵权内容。