北邮复试_2010_哈弗曼树权值计算

题目描述

哈夫曼树，第一行输入一个数n，表示叶结点的个数。需要用这些叶结点生成哈夫曼树，根据哈夫曼树的概念，这些结点有权值，即weight，题目需要输出所有结点的值与权值的乘积之和。

分析：明确计算方法

#include <iostream>
using namespace std;
#include<vector>
#include<queue>
int HarfTree(int value[],int n)
{
    priority_queue < int, vector<int>, greater<int>> q;
    for (int i = 0; i < n; i++)
    {
        q.push(value[i]);
    }
    int a1, a2, ax, sum = 0;
    while (!q.empty())
    {
        a1 = q.top(); q.pop();
        if (q.empty()) break;
        else
        {
            a2 = q.top();
            q.pop();
        }
        ax = a1 + a2;
        sum += ax;
        q.push(ax);  
    }
    return sum;
}
int main()
{
    int n;
    cin >> n;
    int* value = (int*)malloc(sizeof(int*) * n);
    for (int i = 0; i < n; i++)
    {
        cin >> value[i];
    }
    cout << HarfTree(value, n);
}