Python 贪心问题

贪心问题思考的是当下这个步骤的最优解，而且不能影响后续的步骤，在这种情况下可以使用贪心，本篇文章是基于蓝桥杯题目谈判讲解一种贪心思路。

这一题的贪心思路就是每次找到最小的两个部落谈判，这样最后的结果就是花费最少的。

在这里我们要引入python的堆函数heapq函数，这个函数经常用来求动态的最值，因为其中的heapq.heappop(a)可以弹出这个堆中的最小值，而heapq.heappush(a,x)可以往堆中添加元素x。

import os
import sys
import heapq
# 请在此输入您的代码
#读入一个数n和一个数组a
n = int(input())
a = list(map(int,input().split()))
#创建最小堆
heapq.heapify(a)
#记录最后的花费
ans = 0
#当部落数大于1时一直合并
while len(a) >= 2:
  #弹出最小值x和y
  x = heapq.heappop(a)
  y = heapq.heappop(a)
  #把合并后的部落在放回最小堆中
  heapq.heappush(a,x + y)
  #记录合并花费的费用
  ans += x + y
print(ans)

这个贪心思路只能解决这一类取最小值合并的问题，贪心思路不止一种但是核心思路就是取当前最优的选择。