最近看挑战程序设计,就做了书上的题目。
贪心篇的这个题目:
农夫约翰为了修理栏杆,要将一块很长的木板切割成N块。准备切成的木板的长度为L1,L2,,L3,.....LN,未切割前木板的长度恰好为切割后木板长度的总和。
每次切断木板时,需要的开销时这块木板的长度,例如长度为21的木板要切成长度为5,8,8的三块木板,长21的木板切成长为13和8的板时,开销为21.再
将长度为13的板切成长度为5和8的板时,开销时13,于是合计开销就是34.按照题目要求将木板切割完最小的开销时多少。
限制条件:1<=N<=20000
0<=Li<=50000
一开始自己的想法就是切的时候最好两部分长度差不多,每次都这样切,这样想是很正常的。书上介绍了方法,构建二叉树,则合计的开销就是木板的长度*节点
的深度,看了后发现这就是数据结构中的哈夫曼树。答案就是哈夫曼树的带权路径长度,最开始我把哈夫曼树求了一下,结果N*N的复杂度超时了,又想了想
其实也不用去求哈夫曼树的过程中求答案,就用优先队列模拟一下就可以了。
平时构建哈夫曼树的时候,都是求没有父母的节点中权重最小的两个作为一个节点的左右孩子。然后标记这两个节点有了父母。
到优先队列,直接定义权值小的优先出对,每次出对两个元素,权值相加后再入队,当然最后会只剩一个跟节点,直接让其出对就
好了,其权值不必往答案上累加,因为会再入队前就累加节点的值。
AC代码:
#include <iostream>
#include <stdio.h>
#include <queue>
using namespace std;
const int maxn = 20010;
int a[maxn],n;
struct node
{
long long weight;
friend bool operator<(node n1,node n2)
{
return n1.weight>n2.weight;
}
};
long long solve()
{
priority_queue<node>qu;
node cur,min1,min2; ///min1最小值,min2次小值
for(int i = 0; i < n; i++)
{
cur.weight = a[i];
qu.push(cur);
}
long long ans=0;
while(!qu.empty())
{
min1 = qu.top();
qu.pop();
if(!qu.empty())
{
min2 = qu.top();
qu.pop();
cur.weight = min1.weight + min2.weight;
ans += cur.weight;
qu.push(cur);
}
}
return ans;
}
int main()
{
while(~scanf("%d",&n))
{
for(int i = 0; i < n; i++)
scanf("%d",&a[i]);
printf("%lld\n",solve());
}
return 0;
}