哈夫曼树——洛谷 P2168 荷马史诗

来源:互联网 发布:手机淘宝贷款 编辑:程序博客网 时间:2024/05/22 06:21

https://daniu.luogu.org/problem/show?pid=2168
哈夫曼树话说是初赛的知识;
想当年prayer教初赛也是很怀念的呢;

有 N 个数Ai,每次可以挑两个数字 u 和 v直到只留留下
一个,花费 u + v 的代价删除 u 和 v 并加入 u + v.
|
等价描述:
!|
给每个数字分配一个二进制串串,要求二进制串串没有一
个是另一个的前缀,要求A[i]*Len[i]之和最小

为什么呢?
比如我们有4个单词,出现次数是1 2 3 4;
这里写图片描述

取两个最小的;
这里写图片描述
再重复
这里写图片描述
这里写图片描述
那么最后我们就可以统计答案;

这里写图片描述
所以我们用堆去模拟这个过程;
就好了;
如果进制是k,点数是n;

先添加若干个0直到数字个数满足(N-1)%(K-1)=0

显然;

#include<bits/stdc++.h>#define Ll long longusing namespace std;const Ll N=1e5+5;struct di{    Ll v,deep;    bool operator <(const di &a)const{        if(v!=a.v)return v>a.v;        return deep>a.deep;    }}d[N*10];priority_queue<di>Q;Ll n,m,x,y,z,w,ans;int main(){    scanf("%lld%lld",&n,&m);    for(Ll i=1;i<=n;i++)scanf("%lld",&d[i].v),Q.push(d[i]);    w=n;    while((w-1)%(m-1))Q.push(d[++w]);    for(Ll p=(w-1)/(m-1);p;p--){        w++;        for(Ll i=1;i<=m;i++){            di k=Q.top();Q.pop();            d[w].v+=k.v;            d[w].deep=max(d[w].deep,k.deep);        }        d[w].deep++;        ans+=d[w].v;        Q.push(d[w]);    }    printf("%lld\n%lld",ans,d[w].deep);}
原创粉丝点击