LA 6533 Inverting Huffman 構造+貪心

題意：給定哈夫曼樹的n個葉子節點距離根的距離，求文本至少需要多少個字符可以建出這樣的哈夫曼樹

思路：策略：對於第i層的葉子節點，賦值爲i+1層的節點中權值最大的點這種情況下字符數最少。詳見代碼：

/*********************************************************
  file name: LA6533.cpp
  author : kereo
  create time:  2015年02月07日 星期六 22時05分45秒
*********************************************************/
#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
#include<set>
#include<map>
#include<vector>
#include<stack>
#include<cmath>
#include<string>
#include<algorithm>
using namespace std;
typedef long long ll;
const int sigma_size=26;
const int N=100+50;
const int MAXN=100000+50;
const int inf=0x3fffffff;
const double eps=1e-8;
const int mod=1000000000+7;
#define L(x) (x<<1)
#define R(x) (x<<1|1)
#define PII pair<int, int>
#define mk(x,y) make_pair((x),(y))
int n;
ll step;
int a[N];
priority_queue<ll>Q1,Q2;
void init(){
    while(!Q1.empty())
        Q1.pop();
    while(!Q2.empty())
        Q2.pop();
    step=1;
    memset(a,0,sizeof(a));
}
int main(){
    while(~scanf("%d",&n)){
        init();
        for(int i=1;i<=n;i++){
            int x;
            scanf("%d",&x);
            a[x]++;
        }
        ll ans=0;
        for(int i=50;i>=1;i--){
            if(a[i]){
                while(a[i]){
                    Q1.push(step);
                    ans+=step;
                    a[i]--;
                }
            }
            while(Q1.size()>1){
                step=max(step,Q1.top());
                ll tmp=Q1.top(); Q1.pop(); //要long long
                step=max(step,Q1.top());
                tmp+=Q1.top(); Q1.pop();
                Q2.push(tmp);
            }
            if(Q1.size())
                step=max(step,Q1.top());
            while(Q2.size()){
                Q1.push(Q2.top());
                Q2.pop();
            }
        }
        printf("%lld\n",ans);
    }
	return 0;
}