Problem C: 哈夫曼树

Problem C: 哈夫曼树

Time Limit: 1 Sec  Memory Limit: 32 MB
Submit: 64  Solved: 32
[Submit][Status][Web Board]

Description

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

Input

输入有多组数据。
每组第一行输入一个数n，接着输入n个叶节点（叶节点权值不超过100，2<=n<=1000）。

Output

输出权值。

Sample Input

2
2 8 
3
5 11 30 

Sample Output

10
62

把数据保存到数组中，然后直接sort排序，取最小的2个相加再放回数组中，原来的2个数清除，再sort排序，循环下去，最后剩余1个数，即为最后的结果

#include <bits/stdc++.h>
using namespace std;
int main()
{
    int a[1005],n;
    long long int x;
    while(cin>>n)
    {
        int m=n;
        for(int i=0;i<n;i++)cin>>a[i];
        x=0;
        while(n!=1)
        {
            sort(a,a+m);
            x=x+a[0]+a[1];
            n--;
            a[0]=a[0]+a[1];
            a[1]=999999;
        }
        cout<<x<<endl;
    }
    return 0;
}