poj3253:[url]http://poj.org/problem?id=3253[/url],简单的哈夫曼问题
[size=large] 注意此题的数值范围,最后结果用unsigned int或long保存,不然会数据会溢出,另外,注意n=1时,花费为0。[/size]
[size=large]这里优先队列的用法,可以设置比较函数,重载()运算符,还有一种用来比较结构体的方法,重载>操作符。[/size]
#include <iostream>
#include <fstream>
#include <queue>
using namespace std;
struct cmp
{
bool operator() (unsigned int x,unsigned int y)
{
return x>y;
}
};
priority_queue<unsigned int,vector<unsigned int>,cmp> q;
unsigned int ans=0;
unsigned int total=0;
void doit()
{
unsigned int sum=0;
while(!q.empty())
{
sum+=q.top();
q.pop();
if(!q.empty())
{
sum+=q.top();
q.pop();
if(sum!=total)
q.push(sum);
}
ans+=sum;
sum=0;
}
}
int main()
{
unsigned int n;
// ifstream cin("1.txt");
cin>>n;
if(1==n)
{
ans=0;
}
else
{
for(unsigned int i=0;i<n;i++)
{
unsigned int temp;
cin>>temp;
q.push(temp);
total+=temp;
}
doit();
}
cout<<ans<<endl;
return 0;
}
[size=large] 注意此题的数值范围,最后结果用unsigned int或long保存,不然会数据会溢出,另外,注意n=1时,花费为0。[/size]
[size=large]这里优先队列的用法,可以设置比较函数,重载()运算符,还有一种用来比较结构体的方法,重载>操作符。[/size]
struct node
{
int x, y;
friend bool operator < (node a, node b)
{
return a.x > b.x; //结构体中,x小的优先级高
}
};
priority_queue<node>q;//定义方法