1. 程式人生 > >bzoj 4198 [ Noi 2015 ] 荷馬史詩 —— 哈夫曼編碼(k叉哈夫曼樹)

bzoj 4198 [ Noi 2015 ] 荷馬史詩 —— 哈夫曼編碼(k叉哈夫曼樹)

log mes com can rest opera 編碼 type pro

題目:https://www.lydsy.com/JudgeOnline/problem.php?id=4198

第一次寫哈夫曼樹!看了很多博客。

哈夫曼樹 & 哈夫曼編碼:https://www.cnblogs.com/xidongyu/p/6031518.html

這道題:http://www.cnblogs.com/LadyLex/p/7503173.html

滿 k 叉樹節點數的判斷:https://blog.csdn.net/cqbzwja/article/details/46974241

           https://blog.csdn.net/morestep/article/details/47665135

真美啊!

代碼如下:

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<queue>
using namespace std;
typedef long long ll;
int n,k;
struct N{
    ll w,dep;
    N(ll w=0,int d=0):w(w),dep(d) {}
    bool operator < (const N &y) const
    {
        return
w==y.w?dep>y.dep:w>y.w; } }ans; priority_queue<N>q; int main() { scanf("%d%d",&n,&k); ll x; for(int i=1;i<=n;i++) { scanf("%lld",&x); q.push(N(x,1)); } while(k>2&&n%(k-1)!=1)q.push(N(0,1)),n++; while(q.size()>1) { N nw; nw.w
=nw.dep=0; for(int i=1;i<=k;i++) { N t=q.top(); q.pop(); nw.w+=t.w; nw.dep=max(nw.dep,t.dep+1); } ans.w+=nw.w; ans.dep=max(ans.dep,nw.dep); q.push(nw); } printf("%lld\n%d\n",ans.w,ans.dep-1); return 0; }

bzoj 4198 [ Noi 2015 ] 荷馬史詩 —— 哈夫曼編碼(k叉哈夫曼樹)