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 long
using 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);
}