[NOI2015]荷马史诗

一、题目

点此看题

二、解法

这道题正解是哈夫曼树，我们定义每个元素有两个关键字，分别是 出现次数和树的高度（把每个元素看成一颗树），自小到大排序，然后取出最小的$k$个元素，将它们合并为一颗树，合并后的出现次数是这$k$个元素出现次数之后，深度是最大深度加1，塞回数据结构，下面给出贪心的正确性证明：

• 出现次数最小的元素一定在哈夫曼树的最底端，否则我们可以通过交换操作使树的形态更优。
• 合并之后塞回去构造的哈夫曼树和不合并的哈夫曼树是一颗树，这个可由出现次数最小的元素在最底端推得，所以这个贪心的本质是缩小了问题的规模，缩小到了一个点就可以知道问题的答案，故正确性显然。

每次合并操作会消耗掉$k-1$个点，最后要剩下一个点，所以我们要填空，注意如果无空可填不要多填。

#include <cstdio>
#include <queue>
#define int long long
using namespace std;
int read()
{
    int num=0,flag=1;char c;
    while((c=getchar())<'0'||c>'9')if(c=='-')flag=-1;
    while(c>='0'&&c<='9')num=(num<<3)+(num<<1)+(c^48),c=getchar();
    return num*flag;
}
int n,m,k,ans;
struct node
{
	int w,h;
	bool operator < (const node &R) const
	{
		if(w!=R.w) return w>R.w;
		return h>R.h;
	}
};priority_queue<node> q;
signed main()
{
	n=read();k=read();
	for(int i=1;i<=n;i++)
		q.push(node{read(),1});
	if((n-1)%(k-1)) m=k-1-(n-1)%(k-1);//填空
	for(int i=1;i<=m;i++)
		q.push(node{0,1});
	m+=n;//填空后点数
	while(m>1)
	{
		int t1=0,t2=0;
		for(int i=1;i<=k;i++)
		{
			t1+=q.top().w;
			t2=max(t2,q.top().h);
			q.pop();
		}
		ans+=t1;//计算贡献
		q.push(node{t1,t2+1});//合并操作
		m-=(k-1);
	}
	printf("%lld\n%lld\n",ans,q.top().h-1);
}