蓝桥杯 哈夫曼树(python)

蓝桥杯 哈夫曼树(python)_第1张图片

样例输入
5
5 3 8 2 9
样例输出
59

思路:根据题目要求,我们首先要知道输入的格式,确定之后,按照题目的要求我们可知,是要根据哈夫曼树的构造方法,即寻找两个最小的叶子节点然后组成一个节点,最终形成哈夫曼树的过程。但是题目要求的是输出所有组成节点的值。

所以我们就必须要有一个计数的东西,在按照要求,同下方代码一样,一步一步进行即可。

n = int(input())
#输入整数n
mylist = list(map(int,input().split()))
#输入哈夫曼数
count = 0
#计算值
mylist.sort()
#从小到大排序
for i in range(len(mylist)-1):
    mylist.append(mylist[0]+mylist[1])
#找第一和第二个最小的数相加,并添加进原列表
    count += mylist[1]+mylist[0]
#更新值
    mylist.pop(0)
    mylist.pop(0)
#因为是两项相加,所有pop两次
    mylist.sort()
#而后在排序
print(count)

你可能感兴趣的:(排序算法,数据结构,算法,蓝桥杯)