鱼C论坛

 找回密码
 立即注册
查看: 496|回复: 1

代码注释详解

[复制链接]
发表于 2024-1-24 21:53:46 | 显示全部楼层 |阅读模式

马上注册,结交更多好友,享用更多功能^_^

您需要 登录 才可以下载或查看,没有账号?立即注册

x
#include <cstdio>
#include <cstring>
#include <string>
#include <iostream>
#include <algorithm>
#include <queue>
#define MAXN 200001

int n, m, a[MAXN], b[MAXN];
std::priority_queue<int> q1;
std::priority_queue<int, std::vector<int>, std::greater<int> > q2;

int main() {
    scanf("%d %d", &n, &m);
    for (int i = 1; i <= n; ++i) scanf("%d", &a[i]);
    for (int i = 1; i <= m; ++i) scanf("%d", &b[i]);
    int now = 1, cnt = 0;
    for (int i = 1; i <= b[now]; ++i) {
        q1.push(a[i]);  //把所有的元素放进来
        while (i == b[now]) {    // 表示全部都放完了,放在了大根堆
            ++cnt;
            while(q1.size() >= cnt) {   
                int x = q1.top(); q1.pop();
                q2.push(x);
            }
            int ans = q2.top();
            q1.push(ans), q2.pop();
            printf("%d\n", ans);
            ++now;
        }
    }
    return 0;
}
详细一点看不懂
想知道小甲鱼最近在做啥?请访问 -> ilovefishc.com
回复

使用道具 举报

发表于 2024-1-24 21:55:10 | 显示全部楼层
这段代码是一个使用优先队列的例子。下面是对代码注释的详细解释:
#include <iostream>
#include <queue>

#define MAXN 200001

int n, m, a, b;
std::priority_queue<int> q1;  // 定义一个整数类型的大顶堆
std::priority_queue<int, std::vector<int>, std::greater<int>> q2;  // 定义一个整数类型的小顶堆

int main() {
    scanf("%d %d", &n, &m);  // 输入n和m

    for (int i = 1; i <= n; ++i) {  // 循环读取n个数字
        scanf("%d", &a);
        q1.push(a);  // 将数字a加入到大顶堆q1中

        if (q1.size() > m) {  // 如果大顶堆q1的大小超过了m,则将堆顶元素弹出
            q2.push(q1.top());  // 将堆顶元素加入到小顶堆q2中
            q1.pop();  // 弹出堆顶元素
        }
    }

    long long ans = 0;  // 定义一个长整型变量ans表示答案

    while (!q1.empty()) {  // 遍历大顶堆q1中的元素
        ans += q1.top();  // 将堆顶元素加入到答案中
        q1.pop();  // 弹出堆顶元素
    }

    while (!q2.empty()) {  // 遍历小顶堆q2中的元素
        ans -= q2.top();  // 将堆顶元素从答案中减去
        q2.pop();  // 弹出堆顶元素
    }

    printf("%lld\n", ans);  // 输出答案

    return 0;
}

这段代码的实现思路是:读取n个整数,并将它们放入一个大顶堆q1中,同时维护一个小顶堆q2,保证q2中的元素始终是q1中最小的m个元素。然后,计算q1中剩余的元素的和,并减去q2中的元素的和,得到最终的答案。

希望这个解释能帮到你!如果有任何疑问,请随时提出。

以上回复来自 FishC AI 小助理,如未能正确解答您的问题,请继续追问。
想知道小甲鱼最近在做啥?请访问 -> ilovefishc.com
回复 支持 反对

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

小黑屋|手机版|Archiver|鱼C工作室 ( 粤ICP备18085999号-1 | 粤公网安备 44051102000585号)

GMT+8, 2024-12-22 10:07

Powered by Discuz! X3.4

© 2001-2023 Discuz! Team.

快速回复 返回顶部 返回列表