codeforces-884D Boxes And Balls(思维、三叉哈夫曼树)

题意：

有n种球和n个盒子，给n个数代表第i种颜色的球有多少个，初始所有球都在第一个盒子中，现让你将所有的第i颜色的球全部放在第i个盒子中，每次只能选择一个非空盒子，然后将其中的所有球全部拿出来，然后再选k个空盒子，将所有拿出来的球全部任意分配至这k个盒子中(k只能为2和3)，每次的消耗是从一个盒子中拿出球的个数。求将所有球归位的最小的消耗和。

思路：

直接能想到每次贪心前两大的数量，但是不能造成最优。我们可以逆向思考，将所有归位的球最小消耗地合并至第一个盒子中，每次的操作消耗是k个盒子球数量的和。那么就转化为了构造最小根的树，即哈夫曼树。而k可能为2，可能为3，那么肯定是直接合并3个的消耗更少，所以我们要最多的使用k=3。当奇数时，能构成一个完成的3叉哈夫曼树，当偶数时，需要有一个子树为2叉。那么用哪个呢，肯定是最小的两个节点。(即使强制构成三叉，在构到根时，会不断的回调此叉外最大的节点，最终也是回溯到最小的两个节点。)

代码：

#include <bits/stdc++.h>
#define ll long long
using namespace std;
int n;
ll x, ans;
priority_queue<ll, vector<ll>, greater<ll> > q;
int main()
{
    scanf("%d", &n);
    ans = 0;
    for(int i = 1; i <= n; ++i)
    {
        scanf("%lld", &x);
        q.push(x);
    }
    if(!(n&1))
    {
        x = q.top(); q.pop();
        x += q.top(); q.pop();
        q.push(x);
        ans += x;
    }
    while(q.size() > 1)
    {
        x = q.top(); q.pop();
        x += q.top(); q.pop();
        x += q.top(); q.pop();
        q.push(x);
        ans += x;
    }
    printf("%lld\n", ans);
    return 0;
}

继续加油~