给出一数组pi=[p0, p1, ..., pn-1],用这个数组构造成一个Huffman树，过程如下: 1. 找到pi中最小的两个数,设为pa和pb,将pa和pb从pi中删除掉,然后将它们的和加入到pi中。这个过程的费用记为pa + pb。 2. 重复步骤1,直到pi中只剩下一个数。 在上面的操作过程中,把所有的费用相加,就得到了构造Huffman树的总费用。 请你编写一个程序，:对于给定的一个数组，输出构造Huffman树的总费用。 【输入描述】 输入两行 第一行输入整数n,表示数组的长度。 第二行n个整数,表示构成数组的元素，用空格隔开。

import heapq

n = int(input())
nums = list(map(int, input().split()))

# 将数组中的数字转化为节点对象
nodes = [Node(num) for num in nums]

# 构造Huffman树
heapq.heapify(nodes)
while len(nodes) > 1:
    node1 = heapq.heappop(nodes)
    node2 = heapq.heappop(nodes)
    new_node = Node(node1.val + node2.val)
    new_node.left = node1
    new_node.right = node2
    heapq.heappush(nodes, new_node)

# 计算总费用
root = nodes[0]
total_cost = dfs(root, 0)

print(total_cost)

其中，Node类表示Huffman树中的节点，dfs函数用于遍历Huffman树并计算总费用。

class Node:
    def __init__(self, val):
        self.val = val
        self.left = None
        self.right = None

def dfs(node, depth):
    cost = 0
    if node.left:
        cost += dfs(node.left, depth+1)
    if node.right:
        cost += dfs(node.right, depth+1)
    if not node.left and not node.right:
        cost = node.val * depth
    return cost