Skip to content

Commit c307a60

Browse files
authored
Update lz78.py
fixed errors
1 parent 1afdf0f commit c307a60

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

compression/lz78.py

+4-3
Original file line numberDiff line numberDiff line change
@@ -62,9 +62,9 @@ def compress(self, text: str) -> list[Token]:
6262
if phrase not in phrase_dict:
6363
phrase_dict[phrase] = str(code)
6464
if len(phrase) == 1:
65-
tokens.append(Token("0", phrase))
65+
tokens.append(Token(0, phrase))
6666
else:
67-
tokens.append(Token(phrase_dict[phrase[:-1]], phrase[-1]))
67+
tokens.append(Token(int(phrase_dict[phrase[:-1]]), phrase[-1]))
6868
code += 1
6969
phrase = ""
7070
return tokens
@@ -89,7 +89,8 @@ def decompress(self, tokens: list[Token]) -> str:
8989
... Token(0, 'c'), Token(2, 'a'), Token(5, 'b'), Token(1, 'a')])
9090
'ababcbababaa'
9191
>>> lz78_compressor.decompress([Token(0, 'a'), Token(1, 'c'), Token(1, 'a'),
92-
... Token(0, 'c'), Token(1, 'b'), Token(4, 'a'), Token(0, 'b'), Token(3, 'a')])
92+
... Token(0, 'c'), Token(1, 'b'), Token(4, 'a'),
93+
... Token(0, 'b'), Token(3, 'a')])
9394
'aacaacabcabaaa'
9495
"""
9596

0 commit comments

Comments
 (0)