[NOI2015]荷马史诗

题目:洛谷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);
}

 

posted @ 2018-06-08 19:12  Mrsrz  阅读(197)  评论(0编辑  收藏  举报