7-3 修理牧场,关于哈夫曼树,以及优先队列

最近做了一道题,题目如下= =

农夫要修理牧场的一段栅栏,他测量了栅栏,发现需要N块木头,每块木头长度为整数L​i​​个长度单位,于是他购买了一条很长的、能锯成N块的木头,即该木头的长度是L​i​​的总和。
但是农夫自己没有锯子,请人锯木的酬金跟这段木头的长度成正比。为简单起见,不妨就设酬金等于所锯木头的长度。例如,要将长度为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

拿到题目的时候,想的是构造一个哈夫曼树。大概就一个个慢慢叠起来吧
然后我就真的构造了一棵树。。。。

#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;
}

优先队列真好用。。。。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值