4198: [Noi2015]荷马史诗贪心+优先队列

最新推荐文章于 2022-09-17 22:02:54 发布

原创最新推荐文章于 2022-09-17 22:02:54 发布 · 600 阅读

0 ·

CC 4.0 BY-SA版权

My Code 专栏收录该内容

349 篇文章

订阅专栏

本文介绍了一个关于K叉哈弗曼树编码的实现案例。作者虽然自称不太了解哈弗曼编码的具体原理，但通过这个简单的实现分享了如何构建和使用K叉哈弗曼树来解决实际问题。

其实我并不太懂Huffman编码的原理什么的。。只是套K叉哈弗曼树罢了。。
~~然而由于初赛一直随便水所以哈弗曼树是什么我也不太懂只知道是贪心。。~~

#include<bits/stdc++.h>
#define ll long long 
using namespace std;
int n,k;
ll ans;
priority_queue<pair<ll,int>,vector<pair<ll,int> >,greater<pair<ll,int> > > q;
inline ll read()
{
    ll a=0,f=1; char c=getchar();
    while (c<'0'||c>'9') {if (c=='-') f=-1; c=getchar();}
    while (c>='0'&&c<='9') {a=a*10+c-'0'; c=getchar();}
    return a*f;
}
int main()
{
    n=read(); k=read();
    for (int i=1;i<=n;i++) q.push(make_pair(read(),0));
    while (k!=2&&(n-1)%(k-1)) n++,q.push(make_pair(0,0));
    while (q.size()>1)
    {
        ll x=0; int y=0;
        for (int i=1;i<=k;i++)
            x+=q.top().first,y=max(y,q.top().second),q.pop();
        ans+=x; q.push(make_pair(x,y+1));
    }
    cout << ans << endl << q.top().second << endl;
    return 0;
}