51nod 1117 聪明的木匠 （贪心、优先队列）

1117 聪明的木匠 

题目来源： 河北大学算法艺术协会

基准时间限制：1 秒 空间限制：131072 KB 分值: 20  难度：3级算法题

 收藏

 关注

一位老木匠需要将一根长的木棒切成N段。每段的长度分别为L1,L2,......,LN（1 <= L1,L2,…,LN <= 1000，且均为整数）个长度单位。我们认为切割时仅在整数点处切且没有木材损失。

木匠发现，每一次切割花费的体力与该木棒的长度成正比，不妨设切割长度为1的木棒花费1单位体力。例如：若N=3，L1 = 3,L2 = 4,L3 = 5，则木棒原长为12，木匠可以有多种切法，如：先将12切成3+9.，花费12体力，再将9切成4+5，花费9体力，一共花费21体力；还可以先将12切成4+8，花费12体力，再将8切成3+5，花费8体力，一共花费20体力。显然，后者比前者更省体力。

那么，木匠至少要花费多少体力才能完成切割任务呢？

Input

第1行：1个整数N(2 <= N <= 50000)
第2 - N + 1行：每行1个整数Li(1 <= Li <= 1000)。

Output

输出最小的体力消耗。

Input示例

3
3
4
5

Output示例

19

思路：本题的思想就是按照构造哈夫曼树的思想往下进行。给出N个点，每个点都自带权值，每次都挑选队列中权值最小的两个点构成一个父节点，然后去掉这两个点，由父节点代之，依次向下进行，直到队列中只剩下一个节点为止。由于测试数据较多，不适合在for循环中使用sort函数，会超时，所以借助优先队列这个数据结构来完成对队列节点的排序。

#include<bits/stdc++.h>
using namespace std;
int main()
{
    priority_queue<int,vector<int>,greater<int> >Q;
    ///注意greater<int> >，最后一个'>'符号前要有空格
    int n,e,sum=0;
    cin>>n;
    for(int i=0;i<n;i++)
    {
        cin>>e;
        Q.push(e);
    }
    while(Q.size()>1)
    {
        int a=Q.top();
        Q.pop();
        int b=Q.top();
        Q.pop();
        sum+=a+b;
        Q.push(a+b);
    }
    cout<<sum<<endl;
    return 0;
}