最近做了一道题,题目如下= =
农夫要修理牧场的一段栅栏,他测量了栅栏,发现需要N块木头,每块木头长度为整数Li个长度单位,于是他购买了一条很长的、能锯成N块的木头,即该木头的长度是Li的总和。
但是农夫自己没有锯子,请人锯木的酬金跟这段木头的长度成正比。为简单起见,不妨就设酬金等于所锯木头的长度。例如,要将长度为20的木头锯成长度为8、7和5的三段,第一次锯木头花费20,将木头锯成12和8;第二次锯木头花费12,将长度为12的木头锯成7和5,总花费为32。如果第一次将木头锯成15和5,则第二次锯木头花费15,总花费为35(大于32)。
请编写程序帮助农夫计算将木头锯成N块的最少花费。
输入格式:
输入首先给出正整数N(≤104),表示要将木头锯成N块。第二行给出N个正整数(≤50),表示每段木块的长度。
输出格式:
输出一个整数,即将木头锯成N块的最少花费。
输入样例:
8
4 5 1 2 1 3 1 1
输出样例:
49
拿到题目的时候,想的是构造一个哈夫曼树。大概就一个个慢慢叠起来吧
然后我就真的构造了一棵树。。。。
#include<iostream>
#include<bits/stdc++.h>
using namespace std;
struct Tree
{
int weight; //权值
int left,right; //左右孩子
int use;
};
int cmp(Tree a,Tree b) //sort的判断
{
return a.weight>b.weight;
}
int main()
{
Tree tree[20008];
int num;
cin>>num;
int m = (num * 2) - 1;
int treeweight;
for(int i=0;i<num;i++) //输入开始的树。。
{
cin>>treeweight;
tree[i].weight = treeweight;
tree[i].left = -1;
tree[i].right = -1;
tree[i].use = 0; //0是没有使用
}
int minn; //当前最小数
int min_point; //当前最小数的下标
for(int i=num;i<m;i++)
{
//sort(tree,tree+num,cmp);
tree[i].weight = 0;
tree[i].use = 0;
//左孩子
minn = 0xfffffff;
for(int j=0;j<i;j++)
{
if(tree[j].use == 0 )
{
if(tree[j].weight < minn)
{
minn = tree[j].weight;
min_point = j;
}
}
}
tree[i].left = min_point;
tree[min_point].use = 1;
tree[i].weight += tree[min_point].weight;
//右孩子
minn = 0xfffffff;
for(int j=0;j<i;j++)
{
if(tree[j].use == 0 )
{
if(tree[j].weight < minn)
{
minn = tree[j].weight;
min_point = j;
}
}
}
tree[i].right = min_point;
tree[min_point].use = 1;
tree[i].weight += tree[min_point].weight;
}
int all_weight = 0;
for(int i=num;i<m;i++)
{
all_weight += tree[i].weight;
}
cout<<all_weight;
}
这个做法,拿到了18分,理所当然的在最后一个样例超时了
然后我想了想,决定用数组来模拟一下,以下是我做并且对了的代码
#include<iostream>
#include<bits/stdc++.h>
using namespace std;
int main()
{
int num;
int tree[10008] = {0};
cin>>num;
int money = 0;
for(int i=0;i<num;i++)
{
cin>>tree[i];
}
sort(tree,tree+num); //开始的时候先排一次序
for(int j=0;j<num-1;j++)
{
tree[j+1] += tree[j]; //每次第二个数已经变成了1-2树相加
money += tree[j+1];
for(int i=j+1;i<num-1;i++) //将新生成的树移动到他应在的位置,然后BREAK
//即每一次都是一个递增序列,去前两个合成树。。
{
if(tree[i]>tree[i+1])
{
swap(tree[i],tree[i+1]);
}
else
break;
}
}
cout<<money;
}
也许是因为有break的原因,所以能够通过测试,然后有一天我突然知道了
优先队列!(#include 这是头文件)
priority_queue<int> p; //这是大在上的优先队列
priority_queue<int, vector<int>, greater<int> > p; //小在上的优先队列
所以,理所当然的,我这道题新的做法如下==
#include<iostream>
#include<bits/stdc++.h>
#include<queue>
using namespace std;
int main()
{
priority_queue<int,vector<int>,greater<int> >ques;
int num;
cin>>num;
int need;
for(int i=0;i<num;i++)
{
cin>>need;
ques.push(need); //把数存进去
}
int all_money = 0;
int a,b;
int sum = 0;
while(ques.size()!=1)
{
a = ques.top();
ques.pop();
b = ques.top();
ques.pop();
sum = a + b; //找到最小的两个,相加
all_money += sum;
ques.push(sum); //把新生成的 树? 放入
}
cout<<all_money;
return 0;
}
优先队列真好用。。。。