lsmCC 发表于 2024-1-24 21:53:46

代码注释详解

#include <cstdio>
#include <cstring>
#include <string>
#include <iostream>
#include <algorithm>
#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);
    for (int i = 1; i <= n; ++i) scanf("%d", &a);
    for (int i = 1; i <= m; ++i) scanf("%d", &b);
    int now = 1, cnt = 0;
    for (int i = 1; i <= b; ++i) {
      q1.push(a);//把所有的元素放进来
      while (i == b) {    // 表示全部都放完了,放在了大根堆
            ++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;
}
详细一点看不懂

FishC 发表于 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 小助理,如未能正确解答您的问题,请继续追问。
页: [1]
查看完整版本: 代码注释详解