蓝桥杯真题（Python）每日练Day2

尚棹芸2024-01-21 6:07

题目

题目分析

对于本题首先确定其数据结构为优先队列，即邮费最小的衣服优先寄，算法符合贪心算法。
可以直接使用queue库的PriorityQueue方法实现优先队列。
关于PriorityQueue的使用方法主要有：

python 复制代码

import queue
q = queue.Queue()# 队列
pq = queue.PriorityQueue()# 优先队列
>>> q.put(10)#10入队
>>> q.qsize()#队列大小
1
>>> q.get()#出队并返回队首元素
10
>>> q.empty()#队列是否为空
True

尤其注意使用put()函数时，第一个参数priority值越小优先级越高，也就是对首总是最小值。
其实如果使用手写list的方法使用list的sort()方法也可以实现升、降序排列。
本题题目简短却值得深思，可以把染色过程反过来思考，开始，所有的衣服颜色完全不同，最后染成同一种颜色，显然每次都寄出邮费最便宜的两种颜色的衣服，将他们染成通一种颜色，是最省钱的。可以将已经染成同一颜色的两件衣服逻辑上合并为1件衣服，在每个合并步骤中取最小的两个邮费相加，新的邮费在后面继续累加即可。
整个过程和哈夫曼树的原理很相似，贪心算法中运用较多。

题解

python 复制代码

import queue
pq=queue.PriorityQueue()
n=int(input())
a =list(map(int,input().split()))
for i in range(len(a)):
  pq.put(a[i])
sum=0
while pq.qsize()>1:
  t=pq.get()+pq.get()
  sum+=t
  pq.put(t)
print(sum)