Educational Codeforces Round 31- D. Boxes And Balls

D. Boxes And Balls

time limit per test2 seconds
memory limit per test256 megabytes

题目链接:http://codeforces.com/contest/884/problem/D

Description

Ivan has n different boxes. The first of them contains some balls of n different colors.

Ivan wants to play a strange game. He wants to distribute the balls into boxes in such a way that for every i (1 ≤ i ≤ n) i-th box will contain all balls with color i.

In order to do this, Ivan will make some turns. Each turn he does the following:

Ivan chooses any non-empty box and takes all balls from this box;
Then Ivan chooses any k empty boxes (the box from the first step becomes empty, and Ivan is allowed to choose it), separates the balls he took on the previous step into k non-empty groups and puts each group into one of the boxes. He should put each group into a separate box. He can choose either k = 2 or k = 3.
The penalty of the turn is the number of balls Ivan takes from the box during the first step of the turn. And penalty of the game is the total penalty of turns made by Ivan until he distributes all balls to corresponding boxes.

Help Ivan to determine the minimum possible penalty of the game!

Input

The first line contains one integer number n (1 ≤ n ≤ 200000) — the number of boxes and colors.

The second line contains n integer numbers a1, a2, …, an (1 ≤ ai ≤ 109), where ai is the number of balls with color i.

Output

Print one number — the minimum possible penalty of the game.
Educational Codeforces Round 31- D. Boxes And Balls_第1张图片
Note
In the first example you take all the balls from the first box, choose k = 3 and sort all colors to corresponding boxes. Penalty is 6.

In the second example you make two turns:

  1. Take all the balls from the first box, choose k = 3, put balls of color 3 to the third box, of color 4 — to the fourth box and the rest put back into the first box. Penalty is 14;
  2. Take all the balls from the first box, choose k = 2, put balls of color 1 to the first box, of color 2 — to the second box. Penalty is 5.
    Total penalty is 19.

题目精简之后可以看成一个sum分解成一个数列,每次分解的代价就是当前的sum,其实仔细想想就是一个3叉哈夫曼树,当不能凑成三叉的哈夫曼树的时候可以添加0来凑数(不会改变sum)。


哈夫曼树详解:https://www.cnblogs.com/mcgrady/p/3329825.html


#include
using namespace std;
typedef long long ll;
priority_queue vector,greater > qu;
int main()
{
    int n;
    scanf("%d",&n);
    while(n--)
    {
        ll now;
        scanf("%lld",&now);
        qu.push(now);
    }
    if(qu.size()%2 == 0)
        qu.push(0);
    ll ans = 0,a,b,c;
    while(qu.size() > 1)
    {
        a = qu.top(); qu.pop();
        b = qu.top(); qu.pop();
        c = qu.top(); qu.pop();
        ans += a + b + c;
        qu.push(a+b+c);
    }
    printf("%lld",ans);
    return 0;
}

你可能感兴趣的:(算法刷题,codeforces)