哈夫曼樹——洛谷 P2168 荷馬史詩

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