1. 程式人生 > >修理牧場(優先佇列)

修理牧場(優先佇列)

農夫要修理牧場的一段柵欄,他測量了柵欄,發現需要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 <bits/stdc++.h>

using namespace std;

priority_queue<int,vector<int>,greater<int>> p;//這裡有分號;

int main()
{
    int n,a;
    scanf("%d",&n);
    int first,second;
    int sum=0;
    for(int i=0;i<n;i++)
    {
        scanf("%d",&a);
        p.push(a);
    }
    while(p.size()>1)
    {
        first=p.top();
        p.pop();
        second=p.top();
        p.pop();
        sum+=first+second;
        p.push(first+second);//這裡放進去的是first+second,而不是sum;
    }
    printf("%d\n",sum);
}

優先佇列詳解:

1、優先佇列標頭檔案宣告

#include<queue> 
using namespace std;

這倆缺一不可

2、優先佇列的基本格式

    priority_queue<結構型別> 佇列名;

3、常用優先佇列的格式

//從小到大排序,小的先出,大的後出;
priority_queue <int,vector<int>,greater<int> > q;
//從大到小排序,大的先出,小的後出;
priority_queue <int> q;

例如:

#include<cstdio>
#include<queue>
using namespace std;
priority_queue <int> q;
int main()
{
    q.push(10),q.push(8),q.push(12),q.push(14),q.push(6);
    while(!q.empty())
        printf("%d ",q.top());
    q.pop();
}
//輸出結果:14 12 10 8 6

4、優先佇列的基本操作

q.size();//返回q裡元素個數 
q.empty();//返回q是否為空,空則返回1,否則返回0 
q.push(k);//在q的末尾插入k //末尾
q.pop();//刪掉q的第一個元素 //隊頭
q.top();//返回q的第一個元素 
q.back();//返回q的末尾元素