Python：每日一题之小张的衣服（优先队列、哈夫曼编码）

题目描述

小张买了 n 件白色的衣服，他觉得所有衣服都是一种颜色太单调，希望对这些衣服进行染色，每次染色时，他会将某种颜色的所有衣服寄去染色厂，第 i 件衣服的邮费为 ai​ 元，染色厂会按照小张的要求将其中一部分衣服染成同一种任意的颜色，之后将衣服寄给小张， 请问小张要将 n 件衣服染成不同颜色的最小代价是多少？

输入描述

第一行为一个整数 n ，表示衣服的数量。

第二行包括 n 个整数 a1​,a2​...an​ 表示第 i 件衣服的邮费为 ai​ 元。

（1 ≤ n ≤ 10^5,1 ≤ ai ​≤ 10^9 ）

输出描述

输出一个整数表示小张所要花费的最小代价。

输入输出样例

输入

5
5 1 3 2 1 

输出

25

 思考🤔：

题意：意思是一开始你有一堆白衣服，因为是相同的颜色，全部需要送去染色厂，那你就要把这一堆白衣服全部送过去，然后将其中一部分染成另外的颜色，这个时候你就有两种颜色的衣服，那下次送去染色就可以选择是送去白色的一部分，还是送去你染成的颜色的一部分，直到把所有的衣服染成不同的颜色，问最小代价是多少？

例：输入值为

        5

        5 1 3 2 1

        pq = queue.PriorityQueue()  创建优先级队列，若输入值为5 1 3 2 1 ，则为1 1 2 3 5

1. 两次get  2=1+1        put 后  2 2 3 5 
2. 两次get  4=2+1        put 后   3 4 5 
3. 两次get  7=3+4        put 后   5 7
4. 两次get  12=5+7       put 后  12

     最小代价=2+4+7+12=25

反过来思考，也就是将n种颜色的衣服染成一种颜色（白色）

        例：邮费为5 1 3 2 1

参考代码：

import queue
pq=queue.PriorityQueue()  #创建优先级队列，若输入值为5 1 3 2 1 ，则为1 1 2 3 5
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)