>Description
果子按不同种类分成了不同的堆,需要把所有的果子合成一堆。 每一次把两堆果子合并到一起,消耗的体力等于两堆果子的重量之和。
可以看出,所有的果子经过n-1次合并之后,就只剩下一堆了。多多在合并果子时总共消耗的体力等于每次合并所耗体力之和。
你的任务是设计出合并的次序方案,使多多耗费的体力最少,并输出这个最小的体力耗费值。
例如有3种果子,数目依次为1,2,9。可以先将1、2堆合并,新堆数目为3,耗费体力为3。接着,将新堆与原先的第三堆合并,又得到新的堆,数目为12,耗费体力为12。所以多多总共耗费体力=3+12=15。可以证明15为最小的体力耗费值。
>Input
输入包括两行,第一行是一个整数n(1<=n<=10000),表示果子的种类数。第二行包含n个整数,用空格分隔,第i个整数ai(1<=ai<=20000)是第i种果子的数目。
>Output
输出包括一行,这一行只包含一个整数,也就是最小的体力耗费值。输入数据保证这个值小于2^31。
>Sample Input
3
1 2 9
>Sample Output
15
>解题思路
打出来了好感动
比较简单的堆排序,每次把最小和次小的重量加在一起(ans累加),然后把最小和次小的两个数据合并成一个数据,继续排序。这里用堆排序进行排序。
>代码
#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
int n,a[10000005],t;
long long ans;
void up(int s)
{
while(s!=1&&a[s]<a[s/2])
{
t=a[s/2]; a[s/2]=a[s]; a[s]=t;
s/=2;
}
}
void down(int s)
{
while((a[2*s]!=a[0]&&a[2*s]<a[s])||(a[2*s+1]!=a[0]&&a[2*s+1]<a[s]))
{
int y=2*s;
if(a[y+1]<a[y]) y++;
t=a[s]; a[s]=a[y]; a[y]=t;
s=y;
}
}
int main()
{
memset(a,0x7f,sizeof(a));
scanf("%d",&n);
for(int i=1;i<=n;i++) scanf("%d",&a[i]);
for(int i=2;i<=n;i++) up(i); //在原来数据上建堆
while(n>1)
{
int cyz=a[1]; //最小
a[1]=a[n]; a[n]=a[0];
n--; //把最后一个调到堆顶
down(1); //排序
cyz+=a[1]; //次小
a[1]=cyz; //把两数之和放到堆顶
ans+=cyz; //累加数据
down(1); //排序
}
printf("%lld",ans);
return 0;
}