1. 程式人生 > >合併果子(小根堆 手打)

合併果子(小根堆 手打)

題目描述

在一個果園裡,多多已經將所有的果子打了下來,而且按果子的不同種類分成了不同的堆。多多決定把所有的果子合成一堆。

每一次合併,多多可以把兩堆果子合併到一起,消耗的體力等於兩堆果子的重量之和。可以看出,所有的果子經過 n-1n−1 次合併之後, 就只剩下一堆了。多多在合併果子時總共消耗的體力等於每次合併所耗體力之和。

因為還要花大力氣把這些果子搬回家,所以多多在合併果子時要儘可能地節省體力。假定每個果子重量都為 11 ,並且已知果子的種類 數和每種果子的數目,你的任務是設計出合併的次序方案,使多多耗費的體力最少,並輸出這個最小的體力耗費值。

例如有 33 種果子,數目依次為 11 , 22 , 99 。可以先將 11 、 22 堆合併,新堆數目為 33 ,耗費體力為 33 。接著,將新堆與原先的第三堆合併,又得到新的堆,數目為 1212 ,耗費體力為 1212 。所以多多總共耗費體力 =3+12=15=3+12=15 。可以證明 1515 為最小的體力耗費值。

輸入輸出格式

輸入格式:
共兩行。
第一行是一個整數 n(1\leq n\leq 10000)n(1≤n≤10000) ,表示果子的種類數。

第二行包含 nn 個整數,用空格分隔,第 ii 個整數 a_i(1\leq a_i\leq 20000)a
i
​ (1≤a
i
​ ≤20000) 是第 ii 種果子的數目。

輸出格式:
一個整數,也就是最小的體力耗費值。輸入資料保證這個值小於 2^{31}2
31

輸入輸出樣例

輸入樣例#1: 複製
3
1 2 9
輸出樣例#1: 複製
15
說明

對於30%的資料,保證有 n \le 1000n≤1000 :

對於50%的資料,保證有 n \le 5000n≤5000 ;

對於全部的資料,保證有 n \le 10000n≤10000 。

#include<cstdio>
#include<iostream>
#include<cmath>
#include<cstring>
#include<algorithm>

#define N 10009
using namespace std;
int n,k;
int m;
long ans;

int heap[N];

inline void up(int x)
 {
    while(x>1)
     {
        if(heap[x]<heap[x/2
]) { swap(heap[x],heap[x/2]); x/=2; } else break; } } inline void down(int x) { int l=x*2; while(l<=n) { if(heap[l]>heap[l+1]&&l+1<=n)l++; if(heap[l]<heap[x]) { swap(heap[l],heap[x]); x=l;l=x*2; } else break; } } inline void insert(int x) { heap[++n]=x; up(n); } inline void extract() { heap[1]=heap[n--]; down(1); } inline int get_top() { return heap[1]; } int main() { scanf("%d",&m); for(int i=1;i<=m;i++) { scanf("%d",&k); insert(k); } while(m!=1) { m--; int x1=get_top(); extract(); int x2=get_top(); extract(); ans+=x1+x2; insert(x1+x2); } printf("%ld",ans); }