九度OJ－1107－搬水果

題目描述：
在一個果園裏，小明已經將所有的水果打了下來，並按水果的不同種類分成了若干堆，小明決定把所有的水果合成一堆。每一次合併，小明可以把兩堆水果合併到一起，消耗的體力等於兩堆水果的重量之和。當然經過 n‐1 次合併之後，就變成一堆了。小明在合併水果時總共消耗的體力等於每次合併所耗體力之和。
假定每個水果重量都爲 1，並且已知水果的種類數和每種水果的數目，你的任務是設計出合併的次序方案，使小明耗費的體力最少，並輸出這個最小的體力耗費值。例如有 3 種水果，數目依次爲 1，2，9。可以先將 1，2 堆合併，新堆數目爲3，耗費體力爲 3。然後將新堆與原先的第三堆合併得到新的堆，耗費體力爲 12。所以小明總共耗費體力=3+12=15，可以證明 15 爲最小的體力耗費值。
輸入：
每組數據輸入包括兩行,第一行是一個整數 n(1<=n<=10000),表示水果的種類數，如果 n 等於 0 表示輸入結束，且不用處理。第二行包含 n 個整數，用空格分隔，第 i 個整數(1<=ai<=1000)是第 i 種水果的數目。
輸出：
對於每組輸入，輸出一個整數並換行，這個值也就是最小的體力耗費值。輸入數據保證這個值小於 2^31。
樣例輸入：
3
9 1 2
0
樣例輸出：
15

這個題目本質上是求Huffman樹的內節點的和，Huffman樹仍然是模仿word2vec.

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#define MAX 2147483647 //2^31-1

int n;
int * array;
int min1,min2; //最小值和次小值的指針
int pos1,pos2; //掃描指針
int a;
int cmp(const void * a,const void * b){
    return *(int *)b-*(int *)a;
}

int main(){
    freopen("input.txt","r",stdin);
    while(scanf("%d",&n)!=EOF&&n){
        array = (int *)malloc((2*n-1)*sizeof(int));
        for (int i = 0; i < n; ++i)
            scanf("%d",array+i);
        for (int i = n; i < 2*n-1; ++i)
            array[i]=MAX;
        qsort(array,n,sizeof(int),cmp);
        //for (int i = 0; i < 2*n-1; ++i) printf("%d ",array[i]);
        pos1=n-1;
        pos2=n;
        for (a = 0; a < n-1; ++a)
        {
            if (pos1>=0) //find min1
            {
                if (array[pos1]<array[pos2])
                {
                    min1=pos1;
                    pos1--;
                }else{
                    min1 = pos2;
                    pos2++;
                }

            }else{
                min1=pos2;
                pos2++;
            }

            if (pos1>=0) //find min2
            {
                if (array[pos1]<array[pos2])
                {
                    min2=pos1;
                    pos1--;
                }else{
                    min2 = pos2;
                    pos2++;
                }

            }else{
                min2=pos2;
                pos2++;
            }
            array[n+a]=array[min1]+array[min2]; 
        }
        int sum=0;
        for (int i = n; i < 2*n-1; ++i)
        {
            sum+=array[i];
        }
        printf("%d\n",sum);
    }
}