九度OJ - 1172 - 哈夫曼树

题目描述：

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

输入：

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

输出：

输出权值。

样例输入：

5
1 2 2 5 9
样例输出：
37

来源：

2010年北京邮电大学计算机研究生机试真题

思路：该树非叶节点的权值和＝该哈夫曼树的带权路径长度和。本题就是求所有非叶节点的和。用数组保存输入的数组，（然后：排序后把最小的两个数的和放入数组，然后把那两个小数去掉。）对新数组重新排序，重复以上步骤。

ac代码：

#include<bits/stdc++.h>

using namespace std;
int a[2005];
int main(){
    int fir,end,n,sum;
    while(~scanf("%d",&n)){
        sum = 0;
        for(int i = 0;i < n;i++){
            cin >> a[i]; 
        }
        sort(a,a+n);
        fir = 0;
        end = n-1;
        while(fir != end){
            end++;
            a[end] = a[fir]+a[fir+1];
            sum += a[end];
            fir += 2;
            sort(a+fir,a+end+1);//注意这里别写成a+end
        } 
        cout << sum << endl;
    }
    return 0;
}