Create string_compression.py

yingl · web-flow · commit c9e1085ab928 · 2018-04-08T21:49:29.000+08:00
diff --git a/string_compression.py b/string_compression.py
@@ -0,0 +1,25 @@
+class Solution:
+    """
+    @param str: a string
+    @return: a compressed string
+    """
+    def compress(self, str):
+        # write your code here
+        if not str:
+            return str
+        ret = [str[0]]
+        prev_c = str[0]
+        count = 1
+        for i in range(1, len(str)):
+            if str[i] == prev_c:
+                count += 1
+            else:
+                ret.append('%d' % count)
+                prev_c = str[i]
+                ret.append(prev_c)
+                count = 1
+        ret.append('%d' % count)
+        ret = ''.join(ret)
+        return str if len(ret) >= len(str) else ret
+
+# easy: http://lintcode.com/zh-cn/problem/string-compression/
