POJ3253——哈夫曼思想的优先队列

Question：

给一串数，求出哈夫曼树的总代价

Solution:

贪心的哈夫曼思想，我们利用对来进行优先队列的维护
ps:本题中因为虽然限定了每个木块的长度在int型范围内，但是随着我们不断的加和，该数值会超过int的范围，所以说我们用long long 来八寸最后的统计结果，不然会爆掉

Code:

#include"iostream"
#include"cstdio"
#include"cstdlib"
#define N 21010
using namespace std;

long long heap
;
long long n,heapnum,sum;

void swapk(long long x,long long y)
{
long long k=heap[x];
heap[x]=heap[y];
heap[y]=k;
}

void siftdown(int i)
{
int t=i;
while(i*2<=heapnum)
{
if(heap[i]>heap[i*2]) t=i*2;
else t=i;
if(i*2+1<=heapnum&&heap[i*2+1]<heap[t]) t=i*2+1;
if(i!=t)
{
swapk(i,t);
i=t;
}
else break;
}
}

int main()
{
scanf("%I64d",&n);
for(int i=1;i<=n;i++) scanf("%I64d",&heap[i]);
heapnum=n;
for(int i=n/2;i>=1;i--) siftdown(i);
sum=0;
while(heapnum>1)
{
long long k=heap[1];
heap[1]=heap[heapnum];
heapnum--;
siftdown(1);
k+=heap[1];
heap[1]=k;
siftdown(1);
sum+=k;
}
printf("%I64d\n",sum);
return 0;
}