基础实验4-2.7 修理牧场 (25 分)
这题其实是利用了哈夫曼的思想,每次选出最小的两个数,构造一棵哈夫曼树即可。这里我用了STL中的优先队列,代码少,也好理解,STL是真的强大
#include<iostream>
#include<queue>
using namespace std;
int main()
{
priority_queue<int,vector<int>,greater<int> > q;
int n,x;
cin>>n;
for(int i=0;i<n;++i)
{
cin>>x;
q.push(x);
}
int a,b,sum=0;
while(q.size()>1)
{
a=q.top();
q.pop();
b=q.top();
q.pop();
sum+=a+b;
q.push(a+b);
}
cout<<sum;
}