mirror of
https://github.com/TheAlgorithms/Python.git
synced 2024-11-30 16:31:08 +00:00
50545d10c5
* Removed unused commit * Added wikipedia url * Renamed parameter, changed decoding to use list comprehension
49 lines
1.3 KiB
Python
49 lines
1.3 KiB
Python
# https://en.wikipedia.org/wiki/Run-length_encoding
|
|
|
|
|
|
def run_length_encode(text: str) -> list:
|
|
"""
|
|
Performs Run Length Encoding
|
|
>>> run_length_encode("AAAABBBCCDAA")
|
|
[('A', 4), ('B', 3), ('C', 2), ('D', 1), ('A', 2)]
|
|
>>> run_length_encode("A")
|
|
[('A', 1)]
|
|
>>> run_length_encode("AA")
|
|
[('A', 2)]
|
|
>>> run_length_encode("AAADDDDDDFFFCCCAAVVVV")
|
|
[('A', 3), ('D', 6), ('F', 3), ('C', 3), ('A', 2), ('V', 4)]
|
|
"""
|
|
encoded = []
|
|
count = 1
|
|
|
|
for i in range(len(text)):
|
|
if i + 1 < len(text) and text[i] == text[i + 1]:
|
|
count += 1
|
|
else:
|
|
encoded.append((text[i], count))
|
|
count = 1
|
|
|
|
return encoded
|
|
|
|
|
|
def run_length_decode(encoded: list) -> str:
|
|
"""
|
|
Performs Run Length Decoding
|
|
>>> run_length_decode([('A', 4), ('B', 3), ('C', 2), ('D', 1), ('A', 2)])
|
|
'AAAABBBCCDAA'
|
|
>>> run_length_decode([('A', 1)])
|
|
'A'
|
|
>>> run_length_decode([('A', 2)])
|
|
'AA'
|
|
>>> run_length_decode([('A', 3), ('D', 6), ('F', 3), ('C', 3), ('A', 2), ('V', 4)])
|
|
'AAADDDDDDFFFCCCAAVVVV'
|
|
"""
|
|
return "".join(char * length for char, length in encoded)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
from doctest import testmod
|
|
|
|
testmod(name="run_length_encode", verbose=True)
|
|
testmod(name="run_length_decode", verbose=True)
|