【考研每日一題25】哈夫曼樹（C++）

原題地址：牛客網

題目描述：

哈夫曼樹，第一行輸入一個數n，表示葉結點的個數。需要用這些葉結點生成哈夫曼樹，根據哈夫曼樹的概念，這些結點有權值，即weight，題目需要輸出所有結點的值與權值的乘積之和。

輸入描述:

輸入有多組數據。
每組第一行輸入一個數n，接着輸入n個葉節點（葉節點權值不超過100，2<=n<=1000）。

輸出描述:

輸出權值。

示例1

輸入

5  
1 2 2 5 9

輸出

37

分析：

一道哈夫曼樹簽到題。

代碼：

#include<iostream>
#include<vector>
#include<cmath>
#include<queue>
using namespace std;
int main()
{
    int n;
    while(cin>>n)
    {
        int a;
        priority_queue<int,vector<int>,greater<int> >p;
        for(int i=0;i<n;i++)
        {
            cin>>a;
            p.push(a);
        }
        int m1,m2;
        int sum=0;
        while(p.size()>1)
        {
            m1=p.top();
            p.pop();
            m2=p.top();
            p.pop();
            sum=sum+m1+m2;
            p.push(m1+m2);
        }
        cout<<sum<<endl;
    }
    return 0;
}

2020.4.13