Как я понимаю ваш вопрос, вам нужно добавить len(prefix)
к вашему окончательному print
:
def printHuffTree(huffTree, prefix = ''):
if len(huffTree) == 2:
print huffTree[1], prefix, len(prefix)
Обновление для ответа на новый вопрос
def printHuffTree(huffTree, prefix = '', s=0):
if len(huffTree) == 2:
print huffTree[1], prefix, len(prefix)
return len(prefix)*huffTree[0]
else:
s+= printHuffTree(huffTree[1], prefix + '0')
s+= printHuffTree(huffTree[2], prefix + '1')
return s
Вам нужно проверить это