leetcode443. 压缩字符串

给定一组字符，使用原地算法将其压缩。
压缩后的长度必须始终小于或等于原数组长度。
数组的每个元素应该是长度为1 的字符（不是 int 整数类型）。
在完成原地修改输入数组后，返回数组的新长度。

进阶：
你能否仅使用O(1) 空间解决问题？

把逻辑理顺就很好写了：

class Solution:
    def compress(self, chars):
        """
        :type chars: List[str]
        :rtype: int
        """
        res, count = 0, 1
        for i, char in enumerate(chars):
            if i == 0:
                choose = char
                continue
            if char == choose:
                count += 1
            else:
                chars[res] = choose
                if count != 1:
                    for one in str(count):
                        res += 1
                        chars[res] = one
                res += 1
                choose = char
                count = 1
            if i == len(chars)-1:
                chars[res] = choose
                if count != 1:
                    for one in str(count):
                        res += 1
                        chars[res] = one
        return res + 1