使用python+贪心算法+栈数据结构解决问题

题目

共整理四道类似问题，需要用到贪心算法+栈数据结构，题目链接如下。

1. 谈判
2. 合并果子
3. 分金条的最小花费
4. 重新分装

1.谈判

题目描述
在很久很久以前，有n个部落居住在平原上,依次编号为1到n。第i个部落的人数为t_i。
有一年发生了灾荒。 年轻的政治家小蓝想要说服所有部落共同应对灾荒，他能通过谈判来说服部落进行联合。
每次谈判，小蓝只能邀请两个部落参加，花费的金币数量为两个部落的人数之和,谈判的效果是两个部落联合成-个部落(人数为原来两个部落的人数之和)。

输入描述
输入的第一-行包含一个整数 n,表示部落的数量。
第二行包含n个正整数，依次表示每个部落的人数。
其中，1<n≤1000，1<t_i≤10⁴

输出描述
输出一个整数,示最小花费。

因为谈判后，两个部落的人数之和要作为一个新的部落人数，这个人数会被再加一遍，所以谈判过程中的代价越小越好，所以要选最小值。
选最小值，想到数据结构堆，有大根堆和小根堆，大根堆heappop出来自动为最大值，小根堆heappop出来自动为最小值，且堆其实是二叉树，所以插入元素自动排序的时间复杂度低为O(nlogn)

代码如下：

from heapq import heapify, heappop, heappush
n = int(input())
a = list(map(int, input().split()))
ans = 0
# 把a变为堆
heapify(a)
while len(a) > 1