题目描述
在一个果园里,达达已经将所有的果子打了下来,而且按果子的不同种类分成了不同的堆。
达达决定把所有的果子合成一堆。
每一次合并,达达可以把两堆果子合并到一起,消耗的体力等于两堆果子的重量之和。
可以看出,所有的果子经过 n − 1 n-1 n−1 次合并之后,就只剩下一堆了。
达达在合并果子时总共消耗的体力等于每次合并所耗体力之和。
因为还要花大力气把这些果子搬回家,所以达达在合并果子时要尽可能地节省体力。
假定每个果子重量都为 1 1 1,并且已知果子的种类数和每种果子的数目,你的任务是设计出合并的次序方案,使达达耗费的体力最少,并输出这个最小的体力耗费值。
例如有 3 3 3 种果子,数目依次为 1 , 2 , 9 1,2,9 1,2,9。
可以先将 1 、 2 1、2 1、2 堆合并,新堆数目为 3 3 3,耗费体力为 3 3 3。
接着,将新堆与原先的第三堆合并,又得到新的堆,数目为 12 12 12,耗费体力为 12 12 12。
所以达达总共耗费体力 3 ˉ + 12 = 15 \=3+12=15 3ˉ+12=15。
可以证明 15 15 15 为最小的体力耗费值。
输入格式
输入包括两行,第一行是一个整数 n n n,表示果子的种类数。
第二行包含 n n n 个整数,用空格分隔,第 i i i 个整数 a i a_i ai 是第 i i i 种果子的数目。
输出格式
输出包括一行,这一行只包含一个整数,也就是最小的体力耗费值。
输入数据保证这个值小于 2 31 2^{31} 231。
数据范围
1
≤
n
≤
10000
1 \le n \le 10000
1≤n≤10000,
1
≤
a
i
≤
20000
1 \le a_i \le 20000
1≤ai≤20000
输入样例:
3
1 2 9
输出样例:
15
py代码
写法一:优先队列(堆)
'''
@Project :Algorithm
@File :148. 合并果子.py
@IDE :PyCharm
@Author :猫天意
@Date :2024/8/11 21:24:58
'''
import heapq
n = int(input())
q = list(map(int,input().split()))
res = 0
heapq.heapify(q)
while len(q) > 1:
a = heapq.heappop(q)
b = heapq.heappop(q)
res += a + b
heapq.heappush(q,a+b)
print(res)
写法二(基础写法)
'''
@Project :Algorithm
@File :148. 合并果子.py
@IDE :PyCharm
@Author :猫天意
@Date :2024/8/11 21:24:58
'''
n = int(input())
li = list(map(int, input().split()))
res = 0
while len(li) != 1:
num = 0
a = min(li)
li.remove(a)
b = min(li)
li.remove(b)
res += (a + b)
li.append(a + b)
print(res)