problème python3 résolution: compression chaîne

Titre:
compression de chaîne. En utilisant le nombre de caractères récurrents, écrit de manière à obtenir la compression de la chaîne de base. Par exemple, la chaîne devient aabcccccaaa a2b1c5a3. Si la chaîne « compressé » pas plus courte, la chaîne d' origine est retournée. Vous pouvez supposer la chaîne ne contient que les lettres minuscules (a à z).

Exemple 1:

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

Exemple 2:

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

Conseils:

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

Résolution de problèmes:

class Solution:
    def compressString(self, S: str) -> str:
        arr = []
        curW = None #记录当前字符
        curWNum = 0 #记录当前字符的个数
        for w in S:
            if curW != w:
                if curWNum > 0:
                    arr.append(str(curWNum))
                arr.append(w)
                curW = w
                curWNum = 1
            else:
                curWNum += 1
        if curWNum > 0:
            arr.append(str(curWNum))
        res = ''.join(arr)
        return res if len(res) < len(S) else S