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

最新推荐文章于 2024-11-10 14:06:08 发布

想要AC的dly

最新推荐文章于 2024-11-10 14:06:08 发布

阅读量583

点赞数 5

分类专栏：练习题(记录做题想法) 文章标签： leetcode 算法职场和发展 c++ 广度优先

本文链接：https://blog.csdn.net/m0_67724631/article/details/136262921

版权

练习题(记录做题想法) 专栏收录该内容

87 篇文章 2 订阅

订阅专栏

文章介绍了如何通过层序遍历和优先队列（优先级大顶堆）算法，计算给定二叉树中第k大的层和。当树的层数小于k时，返回-1。

摘要由CSDN通过智能技术生成

题目

题目链接
给你一棵二叉树的根节点 root 和一个正整数 k 。
树中的层和是指同一层上节点值的总和。
返回树中第 k 大的层和（不一定不同）。如果树少于 k 层，则返回 -1 。
注意，如果两个节点与根节点的距离相同，则认为它们在同一层。

示例 1：

输入：root = [5,8,9,2,1,3,7,4,6], k = 2
输出：13
解释：树中每一层的层和分别是：

Level 1: 5
Level 2: 8 + 9 = 17
Level 3: 2 + 1 + 3 + 7 = 13
Level 4: 4 + 6 = 10

第 2 大的层和等于 13 。

示例 2：

输入：root = [1,2,null,3], k = 1
输出：3
解释：最大的层和是 3 。

提示：
树中的节点数为 n
2 <= n <= 10⁵
1 <= Node.val <= 10⁶
1 <= k <= n

思路

今天的题目比较简单，主要就是先求出来整个二叉树的每一层的和，然后再找出这些和里面第 k 大的数就可以了，层序遍历可以使用简单的队列实现广度优先搜索，每一层寻找完毕之后可以使用一个小根堆进行存储或者是一个简单的数组存储，如果是数组的话就需要进行排序，如果是大顶堆就需要每次注意堆里面的元素个数超出 k 的时候把最小的元素踢出去。

代码

class Solution {
public:
    long long kthLargestLevelSum(TreeNode* root, int k) {
        priority_queue<long long,vector<long long>,greater<long long>> pq; 
        queue<TreeNode*> que;
        que.push(root);
        while(!que.empty()) {
            long long sum=0;
            int size=que.size();
            for(int i=0;i<size;++i) {
                auto node=que.front();
                que.pop();
                sum+=node->val;
                if(node->left) que.push(node->left);
                if(node->right) que.push(node->right);
            }
            pq.push(sum);
            if(pq.size()>k) pq.pop();
        }
        if(pq.size()<k)return -1;
        return pq.top();
    }
};