LeetCode-面试题 01.06. 字符串压缩-简单

LeetCode-面试题 01.06. 字符串压缩-简单

字符串压缩。利用字符重复出现的次数，编写一种方法，实现基本的字符串压缩功能。比如，字符串aabcccccaaa会变为a2b1c5a3。若“压缩”后的字符串没有变短，则返回原先的字符串。你可以假设字符串中只包含大小写英文字母（a至z）。

示例1：

 输入："aabcccccaaa"
 输出："a2b1c5a3"

示例2：

 输入："abbccd"
 输出："abbccd"
 解释："abbccd"压缩后为"a1b2c2d1"，比原字符串长度更长。

提示：字符串长度在[0, 50000]范围内。

 

代码：

class Solution:
    def compressString(self, S: str) -> str:
        if not S:
            return S
        cur = S[0]
        tmp, count = 0, 0
        res = ''
        L = len(S)
        for i in range(L):
            if S[i] == cur:
                tmp += 1
            else:
                res += cur + str(tmp)
                count += 2
                cur = S[i]
                tmp = 1
            if i == L-1:
                res += cur + str(tmp)
                count += 2
        return res if count < L else S