LeetCode.2583. 二叉树中的第 K 大层和

题目

2583. 二叉树中的第 K 大层和

分析

这道题其实考察的是二叉树的层序遍历，下面我介绍一个二叉树的层序遍历模版：

    public List<List<Integer>> levelOrder(TreeNode root) {// 记录最终的结果List<List<Integer>> res = new ArrayList<>();if(root == null) return res;// 借助队列Queue<TreeNode> queue = new LinkedList<>();queue.add(root);while(!queue.isEmpty()) {// 每一层元素的个数int size = queue.size();// 存放当前层的元素的集合List<Integer> list = new ArrayList<>();while(size != 0) {size--;// 从队列中取出节点TreeNode node = queue.poll();list.add(node.val);// 将当前节点的左右节点放入队列if(node.left != null) queue.add(node.left);if(node.right != null) queue.add(node.right);}res.add(list);}return res;}

这其实是LeetCode的题目：102. 二叉树的层序遍历，大家可以自己写一下。

回归本题：本题我们只需要知道记录每层的元素之和，放入一个集合list中，然后从小到大排列，然后直接取出来第【list.size() - k】个元素即可。

代码

/*** Definition for a binary tree node.* public class TreeNode {*     int val;*     TreeNode left;*     TreeNode right;*     TreeNode() {}*     TreeNode(int val) { this.val = val; }*     TreeNode(int val, TreeNode left, TreeNode right) {*         this.val = val;*         this.left = left;*         this.right = right;*     }* }*/
class Solution {public long kthLargestLevelSum(TreeNode root, int k) {Queue<TreeNode> queue = new LinkedList<>();if(root == null) return -1;List<Long> list = new ArrayList<>();queue.add(root);int level = 0;while(!queue.isEmpty()) {int size = queue.size();level++;long count = 0;while(size != 0) {size--;TreeNode node = queue.poll();count += node.val;if(node.left != null) queue.add(node.left);if(node.right != null) queue.add(node.right);}list.add(count);}Collections.sort(list);if(k <= list.size()) {return list.get(list.size() - k);}return -1;}
}