海贼oj#287. 合并果子

题目描述
​ 在一个果园里，多多已经将所有的果子打了下来，而且按果子的不同种类分成了不同的堆。多多决定把所有的果子合成一堆。

​ 每一次合并，多多可以把两堆果子合并到一起，消耗的体力等于两堆果子的重量之和。可以看出，所有的果子经过 n−1 次合并之后，就只剩下一堆了。多多在合并果子时总共消耗的体力等于每次合并所耗体力之和。

​ 因为还要花大力气把这些果子搬回家，所以多多在合并果子时要尽可能地节省体力。假定每个果子重量都为 1，并且已知果子的种类数和每种果子的数目，你的任务是设计出合并的次序方案，使多多耗费的体力最少，并输出这个最小的体力耗费值。

​ 例如有 3 种果子，数目依次为 1，2，9。可以先将 1、2 堆合并，新堆数目为 3，耗费体力为 3。接着，将新堆与原先的第三堆合并，又得到新的堆，数目为 12，耗费体力为 12。所以多多总共耗费体力为 3+12=15。可以证明 15 为最小的体力耗费值。

输入
​ 输入包括两行，第一行是一个整数 n(1≤n≤10000)，表示果子的种类数。

​ 第二行包含 n 个整数，用空格分隔，第 i 个整数 ai(1≤ai≤20000) 是第 i 种果子的数目。

输出
​ 输出包括一行，这一行只包含一个整数，也就是最小的体力耗费值。输入数据保证这个值小于 231。

`#include <stdio.h>
#include <stdlib.h>
#include <time.h>

#define MAX_ARR 50

int *getRandomArr(int n){
if(n==0) return NULL;
int *temp=(int *)malloc(sizeof(int)*n);
for(int i=0;i<n;i++){
temp[i]=rand()%100;
}
return temp;
}

void swap(int *a,int *b){
if(*a==*b) return ;
*a=*a^*b;
*b=*a^*b;
*a=*a^*b;
return ;
}

void qsort(int *arr,int head,int tail){
if(head>=tail) return;
int first=head+1,last=tail;
while(1){
while(first<=tail&&arr[first]<arr[head]){
first+=1;
continue;
}
while(arr[last]>arr[head]){
last-=1;
continue;
}
if(first<last) swap(&arr[last],&arr[first]);
else break;
}
first-=1;
swap(&arr[head],&arr[first]);
qsort(arr,head,first-1);
qsort(arr,first+1,tail);
return ;
}

void sort_min(int **arr,int i,int j){
if(i>=j) return ;
int temp=(*arr)[i];
int k=i+1;
for(;temp>(*arr)[k]&&k<=j;k++){
(*arr)[k-1]=(*arr)[k];
}
(*arr)[k-1]=temp;
return ;
}

void buildHaffmanTree(int *arr,int *res,int i,int j){
if(i==j) return ;
int temp=0;
temp=arr[i]+arr[i+1];
*res+=temp;
arr[i+1]=temp;
sort_min(&arr,i+1,j);
for(int k=i+1;k<MAX_ARR;k++){
printf("%d ",arr[k]);
}
printf(“temp=%d “,temp);
printf(“res=%d”,*res);
printf(”\n”);
buildHaffmanTree(arr,res,i+1,j);
return ;
}

int main(void){
srand(time(0));
int *arr=getRandomArr(MAX_ARR);
for(int i=0;i<MAX_ARR;i++){
printf(“%d “,arr[i]);
}
printf(”\n”);
qsort(arr,0,MAX_ARR-1);
for(int i=0;i<MAX_ARR;i++){
printf(“%d “,arr[i]);
}
printf(”\n”);
int res=arr[0];
buildHaffmanTree(arr,&res,0,MAX_ARR-1);
printf(“%d”,res);
return 0;
}`
由于排序算法过于低效，在小数据量时可以运行，大数据量就会超时，该程序带有测试部分，去掉测试部分数据量在50左右可以运行。