来源:点击打开链接
很久很久之前做过这个题,印象中是用优先队列来做,结果一写各种wa了。。。。。。。。。。翻之前的代码库,发现优先队列的定义出现了问题。。
因为数据很大需要每次都选取两个最短的进行拼装,所以用了优先队列,每两个小的构成父节点,然后把父节点放进去再找两个小的接起来。huffmanTree的逆向思维,接到最后那一个就是最后的答案了。
#include <iostream>
#include <queue>
#include <vector>
#include <algorithm>
using namespace std;
int main()
{
long long testcase,tmp;
while(cin>>testcase)
{
long long result=0,mid=0,total=0;
priority_queue<long long,vector<long long>,greater <vector<long long>::value_type> > pack;
while(testcase--)
{
cin>>tmp;
pack.push(tmp);
total+=tmp;
}
while(pack.size()>1)//队列里边还有没用完的木头
{
mid=0;
mid+=pack.top();
pack.pop();
mid+=pack.top();
pack.pop();
pack.push(mid);
result+=mid;
}
cout<<result+pack.top()-total<<endl;
}
return 0;
}
/**************************************
Problem id : SDUT OJ 2848
User name : 冰甲·逐日者
Result : Accepted
Take Memory : 1128K
Take Time : 20MS
Submit Time : 2014-05-05 10:09:59
**************************************/