西電覆試之——真題2011D 哈夫曼樹
1 優先隊列
//基礎知識
priority_queue<int> q;
//push()時間複雜度O(logN)
q.push(3);
//優先隊列沒有front()函數和back()函數(隊列中隊首隊尾),只能用top()訪問隊首元素O(1)
q.top();
//pop()時間複雜度O(logN)
q.pop();
//q.empty() O(1)
q.empty();
//q.size() O(1)
q.size();
1.1 priority_queue內元素優先級的設置
//下面兩種優先隊列的定義是等價的
priority_queue<int> q;
//less<int>表示數字大的優先級大,greater<int>表示數字小的優先級越大
priority_queue<int,vector<int>,less<int>>q;
2 真題
輸入樣例:
8
7 19 2 6 32 3 21 10
輸出樣例:
261
使用優先隊列模擬哈夫曼樹的帶權路徑長度
#include<iostream>
#include<queue>
using namespace std;
priority_queue < int, vector<int>, greater<int>> q;
int main() {
int n;
int temp, x, y, ans = 0;
cin >> n;
for (int i = 0; i < n; i++) {
cin >> temp;
q.push(temp);
}
while (q.size() > 1) {
x = q.top();
q.pop();
y = q.top();
q.pop();
q.push(x + y);
ans += x + y;
}
cout << ans;
return 0;
}