【CF884D】Boxes And Balls

題意：有n個箱子和若幹個球，球的顏色也是1-n，有ai個球顏色為i，一開始所有的球都在1號箱子裏，你每次可以進行如下操作：

選擇1個箱子，將裏面所有的球拿出來，分成k部分（你可以令k=2或3），將每一部分都放到一個空箱子中。花費的代價是這個箱子中球的總數。

現要求你在若幹次操作後，所有顏色為i的球都在i號箱子裏，求最小代價。

n<=200000,ai<=10^9

題解：傻逼題都不會做了。將操作反過來，就變成了將k個箱子合並到一起，這就變成經典的哈夫曼樹了。

k既可以=2又可以=3怎麽辦？如果n是奇數的話，我們可以每次都讓k=3，否則我們就再加入一個空箱子即可。

#include <cstdio>
#include <cstring>
#include <iostream>
#include <queue>
using namespace std;
typedef long long ll;
priority_queue<ll,vector<ll>,greater<ll> > q;
int n;
ll ans;
inline int rd()
{
	int ret=0,f=1;	char gc=getchar();
	while(gc<‘0‘||gc>‘9‘)	{if(gc==‘-‘)	f=-f;	gc=getchar();}
	while(gc>=‘0‘&&gc<=‘9‘)	ret=ret*10+(gc^‘0‘),gc=getchar();
	return ret*f;
}
int main()
{
	n=rd();
	int i;
	ll a;
	for(i=1;i<=n;i++)	q.push(rd());
	if(!(n&1))	q.push(0),n++;
	for(i=n;i!=1;i-=2)	a=q.top(),q.pop(),a+=q.top(),q.pop(),a+=q.top(),q.pop(),q.push(a),ans+=a;
	printf("%lld",ans);
	return 0;
}

【CF884D】Boxes And Balls 哈夫曼樹