1. 程式人生 > >[NOI2015]荷馬史詩

[NOI2015]荷馬史詩

AC #define int() pop ace efi print tro air

題目:洛谷P2168、codevs5501。

題目大意:
有n種單詞,每種單詞有一個出現次數。現在要用k進制字符串代表這些單詞,使得任意一個字符串不是另一個的前綴。問:最短的總長度是多少?最短的單詞長度是多少?
解題思路:
哈夫曼編碼構造過程,k進制就每次取k個合並。
用優先隊列實現即可。
註意:最後合並時可能出現不夠,所以可以一開始增加一些空節點占位。

C++ Code:

#include<bits/stdc++.h>
#include<ext/pb_ds/priority_queue.hpp>
#define LoveLive long long
using namespace std;
__gnu_pbds::priority_queue<pair<LoveLive,int>,greater<pair<LoveLive,int> >,__gnu_pbds::pairing_heap_tag>h;
int n,k;
inline LoveLive readint(){
    int c=getchar();
    LoveLive d=0;
    for(;!isdigit(c);c=getchar());
    for(;isdigit(c);c=getchar())d=(d<<3)+(d<<1)+(c^‘0‘);
    return d;
}
int main(){
    n=readint(),k=readint();
    for(int i=1;i<=n;++i)h.push(make_pair(readint(),1));
    if((n-1)%(k-1))
    for(int i=k-1-(n-1)%(k-1);i;--i)h.push(make_pair(0ll,1));
    LoveLive niconiconi=0;
    for(int tot=n+(((n-1)%(k-1))?k-1-(n-1)%(k-1):0);tot>1;tot-=k-1){
        int mx=0;
        LoveLive sm=0;
        for(int i=1;i<=k;++i){
            pair<LoveLive,int>p=h.top();
            h.pop();
            mx=max(mx,p.second);
            sm+=p.first;
        }
        niconiconi+=sm;
        h.push(make_pair(sm,mx+1));
    }
    printf("%lld\n%d\n",niconiconi,h.top().second-1);
}

[NOI2015]荷馬史詩