7-3 修理牧场 (c++哈夫曼树实现)
7-3 修理牧场
农夫要修理牧场的一段栅栏,他测量了栅栏,发现需要N块木头,每块木头长度为整数Li个长度单位,于是他购买了一条很长的、能锯成N块的木头,即该木头的长度是Li的总和。
但是农夫自己没有锯子,请人锯木的酬金跟这段木头的长度成正比。为简单起见,不妨就设酬金等于所锯木头的长度。例如,要将长度为20的木头锯成长度为8、7和5的三段,第一次锯木头花费20,将木头锯成12和8;第二次锯木头花费12,将长度为12的木头锯成7和5,总花费为32。如果第一次将木头锯成15和5,则第二次锯木头花费15,总花费为35(大于32)。
请编写程序帮助农夫计算将木头锯成N块的最少花费。
输入格式:
输入首先给出正整数N(≤10^4 ),表示要将木头锯成N块。第二行给出N个正整数(≤50),表示每段木块的长度。
输出格式:
输出一个整数,即将木头锯成N块的最少花费。
输入样例:
8
4 5 1 2 1 3 1 1
输出样例:
49
插话
sort(a+1,a+n+1,greater());(从大到小)这个题我傻乎乎的创了一个树来做,其实不需要这么麻烦,将数组使用sort函数排序一下,每次取出两个不为0的最小的,这两个位置要清0,相加后再放入数组,再次使用sort数组排序一下。到最后数组中的值就是需要的
代码
#include <bits/stdc++.h>
using namespace std;
#define max 20001
typedef struct Node{
int parent;
int left,right;
int data;
}LNode;
int a[max];
int n,m,sum=0,num=0;
LNode self[max];
void create();
int find(int *x1,int *x2,int count);
int main(){
cin>>n;
m=2*n-1;
for(int i=1;i<=n;i++){
cin>>a[i];
sum+=a[i];
self[i].data=a[i];
self[i].left=self[i].parent=self[i].right=-1;
}
sort(a+1,a+n+1);
// sort(a+1,a+n+1,greater<int>()); //从大到小
for(int i=n+1;i<=m;i++){
self[i].data=0;
self[i].left=self[i].parent=self[i].right=-1;
}
create();
cout<<num;
return 0;
}
void create(){
int x1,x2;
for(int i=n+1;i<=m;i++){
find(&x1,&x2,i);
self[i].data=self[x1].data+self[x2].data;
self[i].left=x1;
self[i].right=x2;
self[x1].parent=i;
self[x2].parent=i;
num=num+self[i].data;
}
}
int find(int *x1,int *x2,int count){
int pmin=max-1,min=max-1;
self[max-1].data=1000000;
for(int i=1;i<count;i++){
if(self[i].parent==-1&&self[min].data>self[i].data){
pmin=min;
min=i;
}else if(self[i].parent==-1&&self[pmin].data>self[i].data){
pmin=i;
}
}
*x1=min;*x2=pmin;
}
本博客所有文章除特别声明外,均采用 CC BY-NC-SA 4.0 许可协议。转载请注明来自 时间海!
评论