Fence Repair POJ - 3253 最小优先队列，哈弗曼树

首先拿用例来说明一下题意

给个一根长度为21的模板，

目标是把它分为三块，长度分别是8,5,8

切割的时候先把他切割成长度为8和13的，消耗21，

然后再把长度为13的模板切割成长度为5和8的，消耗13

然后总消耗就是21+13=34

求最小总消耗

我用的是哈弗曼树来实现，需要注意的是输入n==1，也就是分成一块的时候，这时候直接弹出队列并输出木板长度，否则根据哈夫曼计算总消耗并输出

下面是代码

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<queue>
using namespace std;

int main()
{
int n;
long long int sum;
priority_queue<int, vector<int>, greater<int> > p;//最小优先队列，默认是最大
while(cin>>n)
{
int x;
sum=0;
for(int i=1; i<=n; ++i)
{
cin>>x;
p.push(x);
}
if(n==1)
{
sum=p.top();
p.pop();
}

else
{
int temp1,temp2;
while(p.size()>1)
{
temp1=p.top();
p.pop();
temp2=p.top();
p.pop();
int  temp;
temp=temp1+temp2;
p.push(temp);
sum+=temp;
}
p.pop();
}
cout<<sum<<endl;
}
}