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

news2024/11/17 15:54:37

题目

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;
    }
}