时间限制(秒) | 内存限制(MB) |
---|---|
1 | 128 |
【题目描述】
在一个果园里,多多已经将所有的果子打了下来,而且按果子的不同种类分成了不同的堆。多多决定把所有的果子合成一堆。每一次合并,多多可以把两堆果子合并到一起,消耗的体力等于两堆果子的重量之和。可以看出所有的果子经过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。
【输入样例】
3
1 2 9
【输出样例】
15
【思路】
不专业请多见谅
这是一道典型的用堆做的题。把输入的值全部放入堆中,堆是自动排好序的(可以上网查资料),每次取出两个最小值合并,把和累计,最后只剩一堆输出结果就可以了。
【源程序】
/*函数堆做法!(简洁易懂)*/
#include<iostream>
#include<queue>//建立函数堆需要的头文件
#include<cstdio>
using namespace std;
const short MAX=10000;
int N,a,answer=0,b;
priority_queue<int,vector<int>,greater<int> >q;//建立小根堆q
int main()
{
scanf("%d",&N);
for(int i=1;i<=N;i++)
scanf("%d",&a),q.push(a);//把a依次放入堆
while(true)
{
b=q.top(),q.pop(),b+=q.top(),q.pop(),answer+=b;//b代表合并的值,top表示堆顶元素,pop表示删除堆顶元素
if(q.empty())//如果堆已经变空了
break;
q.push(b);//否则把合并完的果子放入堆
}
printf("%d",answer);//输出结果
return 0;
}
/*手写堆做法*/
#include<iostream>
#include<cstdio>
using namespace std;
const short MAX=10000;
int N,a,queue[MAX+1],queue_length=0,s,answer=0,b;
void push(int x)//堆的push函数
{
queue[++queue_length]=x,s=queue_length;
while(s>1&&queue[s]<queue[s/2])
swap(queue[s],queue[s/2]),s/=2;
}
void pop()//堆的pop函数
{
queue[1]=queue[queue_length],queue_length--,s=1;
while(s*2+1<=queue_length&&queue[s]>min(queue[s*2],queue[s*2+1]))
{
if(queue[s*2]>queue[s*2+1])
{
if(queue[s]>queue[s*2+1])
swap(queue[s],queue[s*2+1]),s*=2,s++;
}
else
if(queue[s]>queue[s*2])
swap(queue[s],queue[s*2]),s*=2;
}
if(s*2<=queue_length&&queue[s*2]<queue[s])
swap(queue[s],queue[s*2]);
}
int top()//堆的top函数
{
return queue[1];
}
bool empty()//堆的empty函数
{
if(queue_length)
return false;
return true;
}
int main()
{
scanf("%d",&N);
for(int i=1;i<=N;i++)
scanf("%d",&a),push(a);//把a依次放入堆
while(true)
{
b=top(),pop(),b+=top(),pop(),answer+=b;//b代表合并的值,top表示堆顶元素,pop表示删除堆顶元素
if(empty())//如果堆已经变空了
break;
push(b);//否则把合并完的果子放入堆
}
printf("%d",answer);//输出结果
return 0;
}