POJ 3253 - Fence Repai ( 优先队列 )

题意

切割木板, 比如一根长21的木板要切割成5, 8, 8的三块, 每次切割花费的金额为两断的长度. 比如先把21切成16和5, 花费21元, 再把16切成8和8, 花费16元, 总计消费37元. 若先把21切成13和8, 花费21元, 再把13切成8和5, 花费13元, 总计消费34元, 花销更少
现在就要求出最小开销

思路

这道题可以用贪心, 这里主要记录一下白书上提供的优先队列 这个非常好的思路
把切割逆向看作拼接过程, 定义一个从小到大排列的优先队列

priority_queue<int, vector<int>, greater<int> > fen;

每次取出最短的两根木板拼接, 再将这个拼接后的长木板推入优先队列
直到fen.size() == 1, 这时木板被拼接完毕

记得看数据范围需要用long long !!

AC代码

#include <iostream>
#include <algorithm>
#include <cstdio>
#include <cstring>
#include <queue>

using namespace std;
const int maxn = 20000 + 100;
priority_queue<int, vector<int>, greater<int> > fen;
typedef long long LL;
//int s[maxn];

int main()
{
    int n, a;
    while( ~scanf("%d",&n) )
    {
        LL sum = 0;
        while( !fen.empty() ) fen.pop();
        for( int i = 0; i < n; i++ ){
            scanf("%d",&a);
            fen.push(a);
        }
        while( fen.size() > 1 ){
            int l1 = fen.top();
            fen.pop();
            int l2 = fen.top();
            fen.pop();
            sum += l1 + l2;
            fen.push(l1+l2);
        }
        printf("%lld\n",sum);
    }
    return 0;
}