描述
在一个果园里,多多已经将所有的果子打了下来,而且按果子的不同种类分成了不同的堆。多多决定把所有的果子合成一堆。
每一次合并,多多可以把两堆果子合并到一起,消耗的体力等于两堆果子的重量之和。可以看出,所有的果子经过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种果子的数目。
输出格式
输出包括一行,这一行只包含一个整数,也就是最小的体力耗费值。输入数据保证这个值小于2^31。
限制
每个测试点1s
来源
NOIp 2004
思路:
算是简单吧?类似哈夫曼树,每次寻找最小的两堆合并
写了两个版本,一个是优先队列的很短,还有就是手打堆
第一次写这么长的c++心中略j动。掌握好了语法知识,算法的核心思想都是差不多的了
源程序/c++:
/*
作者:olahiuj
题目:p1063 合并果子
*/
#include <stdio.h>
#include <queue>
using namespace std;
priority_queue<int>heap;
int main()
{
int n,x,ans=0;
scanf("%d",&n);
for (int i=1;i<=n;i++)
scanf("%d",&x),heap.push(-x);
int a,b;
for (int i=1;i<n;i++)
{
a=-heap.top();heap.pop();
b=-heap.top();heap.pop();
heap.push(-a-b);
ans+=a+b;
}
printf("%d\n",ans);
return 0;
}
#include <stdio.h>
using namespace std;
int t[10000]={0};
int maxT=0;
int ans=0;
void swap(int &x,int &y)
{
int tmp=x;
x=y;
y=tmp;
}
void insert(int x)
{
maxT++;
int v=maxT;
while (v>1&&t[v/2]>x)
{
t[v]=t[v/2];
v/=2;
}
t[v]=x;
}
void fy(int x)
{
int p;
int l=x*2;
int r=x*2+1;
if (l<=maxT&&t[l]<t[x]) p=l;
else p=x;
if (r<=maxT&&t[r]<t[p]) p=r;
if (x!=p)
{
swap(t[x],t[p]);
fy(p);
}
}
int pop()
{
int g=t[1];
t[1]=t[maxT];
maxT--;
fy(1);
return g;
}
int main()
{
int n,x;
scanf("%d",&n);
for (int i=1;i<=n;i++)
{
scanf("%d",&x);
insert(x);
}
while (maxT>1)
{
int x=pop();
int y=pop();
insert(x+y);
ans+=x+y;
}
printf("%d",ans);
return 0;
}