tasks a-s

Task/Run-length-encoding/Python/run-length-encoding-3.py

@@ -0,0 +1,22 @@
+from re import sub
+
+def encode(text):
+    '''
+    Doctest:
+        >>> encode('WWWWWWWWWWWWBWWWWWWWWWWWWBBBWWWWWWWWWWWWWWWWWWWWWWWWBWWWWWWWWWWWWWW')
+        '12W1B12W3B24W1B14W'
+    '''
+    return sub(r'(.)\1*', lambda m: str(len(m.group(0))) + m.group(1),
+               text)
+
+def decode(text):
+    '''
+    Doctest:
+        >>> decode('12W1B12W3B24W1B14W')
+        'WWWWWWWWWWWWBWWWWWWWWWWWWBBBWWWWWWWWWWWWWWWWWWWWWWWWBWWWWWWWWWWWWWW'
+    '''
+    return sub(r'(\d+)(\D)', lambda m: m.group(2) * int(m.group(1)),
+               text)
+
+textin = "WWWWWWWWWWWWBWWWWWWWWWWWWBBBWWWWWWWWWWWWWWWWWWWWWWWWBWWWWWWWWWWWWWW"
+assert decode(encode(textin)) == textin