荷马史诗(N叉哈弗曼树)

题目：

见https://www.acwing.com/problem/content/151/

思路:

这是一个考察n叉哈弗曼树的题目,详情见https://www.acwing.com/video/128/

代码:

#include<iostream>
#include<algorithm>
#include<queue>
#include<vector>
using namespace std;
typedef long long LL;
typedef pair<LL,int> PLI;
LL res;
int main()
{
    int n,k;
    cin>>n>>k;
    priority_queue<PLI,vector<PLI>,greater<PLI> > heap;//小根堆维护两个关键字,除了出现次数还有深度
    for(int i=1;i<=n;i++)
    {
        LL w;
        cin>>w;
        heap.push({w,0});
    }
    while((n-1)%(k-1)) heap.push({0,0}),n++;//关键步骤
    while(heap.size()>1)
    {
        LL s=0;
        int depth=0;
        for(int i=1;i<=k;i++)
        {
            auto temp=heap.top();
            s+=temp.first;
            depth=max(depth,temp.second);
            heap.pop();
        }
        heap.push({s,depth+1});
        res+=s;
    }
    cout<<res<<endl;
    cout<<heap.top().second<<endl;
    return 0;
}