[NOI2015]荷馬史詩
阿新 • • 發佈:2018-06-08
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]荷馬史詩