POJ 3253 Fence Repair（優先隊列，哈夫曼樹）

標籤：style   class   blog   code   http   tar   

題目

 

//做哈夫曼樹時，可以用優先隊列（誤？）

//這道題教我們優先隊列的一個用法：取前n個數（最大的或者最小的）

 

//哈夫曼樹//64位//逾時->優先隊列，，，，//這道題的優先隊列用於取前2個小的元素#include <iostream>#include<stdio.h>#include<string.h>#include<algorithm>#include<queue>using namespace std;__int64 ans,a[20010];int n;struct cmp{      bool operator ()(__int64 &a,__int64 &b){          return a>b;    }  };priority_queue<__int64,vector<__int64>,cmp> q;void hfm(){        __int64 min1,min2,neww,yi;    while(n>1)    {        yi=0;        min1=q.top();        q.pop();        min2=q.top();        q.pop();        neww=min1+min2;        ans+=neww;        q.push(neww);        n--;    }}int main(){    int i;    while(scanf("%d",&n)!=EOF)    {        while(!q.empty())q.pop();        for(i=0;i<n;i++)        {            scanf("%I64d",&a[i]);            q.push(a[i]);        }        ans=0;        if(n>1)            hfm();        else            ans=a[0];        printf("%I64d\n",ans);    }    return 0;}

View Code