Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 7 additions & 1 deletion contents/huffman_encoding/code/python/huffman.py
Original file line number Diff line number Diff line change
Expand Up @@ -29,11 +29,17 @@ def build_huffman_tree(message):
# insert the new tree there
trees.insert(index, (new_tree, new_weight))

huffman_tree = trees[0][0]
# Return an empty list if the message was empty, else the tree
huffman_tree = [] if not trees else trees[0][0]
return huffman_tree

# constructs the mapping with recursion
def build_codebook(tree, code=''):
# Check whether our tree contains more than 1 element or not
if not tree:
return []
elif type(tree) != list:
return [(tree, '0')]

codebook = []

Expand Down