PTA 7-29 修理牧場（Huffman樹）

數據結構和代碼倉庫
本題考點：

• Huffman 樹

農夫要修理牧場的一段柵欄，他測量了柵欄，發現需要N塊木頭，每塊木頭長度為整數L_i
​​ 個長度單位，於是他購買了一條很長的、能鋸成N塊的木頭，即該木頭的長度是L_i的總和。
但是農夫自己沒有鋸子，請人鋸木的酬金跟這段木頭的長度成正比。為簡單起見，不妨就設酬金等於所鋸木頭的長度。例如，要將長度為20的木頭鋸成長度為8、7和5的三段，第一次鋸木頭花費20，將木頭鋸成12和8；第二次鋸木頭花費12，將長度為12的木頭鋸成7和5，總花費為32。如果第一次將木頭鋸成15和5，則第二次鋸木頭花費15，總花費為35（大於32）。
請編寫程序幫助農夫計算將木頭鋸成N塊的最少花費。

仔細讀題之后本題是考察 Huffman 樹的典型題目，關於Huffman樹的原理回顧之前的文章，這里不再贅述，這里主要是一個優先隊列的應用。
完整代碼如下：

#include <iostream>
#include <queue>
using namespace std;

int n;
priority_queue<int, vector<int>, greater<int>> pq; // 小頂堆

int main()
{
    scanf("%d", &n);
    int temp;
    while (n--)
    {
        scanf("%d", &temp);
        pq.push(temp);
    }
    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);
    }
    printf("%d", ans);
    return 0;
}