合併果子(小根堆 手打)
題目描述
在一個果園裡,多多已經將所有的果子打了下來,而且按果子的不同種類分成了不同的堆。多多決定把所有的果子合成一堆。
每一次合併,多多可以把兩堆果子合併到一起,消耗的體力等於兩堆果子的重量之和。可以看出,所有的果子經過 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);
}