zoj1067 优先队列c++
时间: 2023-07-09 17:40:58 浏览: 127
这是一个使用优先队列(priority queue)的题目,可以使用C++ STL中的优先队列来实现。
以下是样例代码:
```c++
#include <iostream>
#include <queue> // 包含 priority_queue 头文件
using namespace std;
int main() {
int n;
while (cin >> n) {
priority_queue<int, vector<int>, greater<int>> pq; // 定义小根堆
for (int i = 0; i < n; i++) {
int x;
cin >> x;
pq.push(x); // 将元素加入优先队列
}
int ans = 0;
while (pq.size() > 1) { // 只要队列中还有两个及以上元素
int a = pq.top(); pq.pop();
int b = pq.top(); pq.pop();
ans += a + b;
pq.push(a + b); // 新元素入队
}
cout << ans << endl;
}
return 0;
}
```
在这个代码中,我们使用了C++ STL中的优先队列 `priority_queue`,它有三个模板参数:
- `int`:表示队列中存储的元素类型是 `int`。
- `vector<int>`:表示队列内部使用 `vector` 作为基础容器。
- `greater<int>`:表示使用小根堆来存储元素。
在主函数中,我们首先读入元素,然后将它们加入到优先队列中。接着,我们依次取出队列中的两个最小元素,将它们相加并累加到答案中,然后将它们的和作为新元素加入到队列中。最后,当队列中只剩下一个元素时,输出答案。
需要注意的是,在使用小根堆时,我们要将模板参数 `greater<int>` 作为第三个参数传递给 `priority_queue`。如果不指定第三个参数,则默认使用大根堆。
另外,这里使用了 `while (cin >> n)` 的方式来不断读入测试用例,直到遇到输入结束符为止(比如EOF或者Ctrl+Z)。
阅读全文