443. 压缩字符串

给你一个字符数组 chars ，请使用下述算法压缩： 从一个空字符串 s 开始。对于 chars 中的每组 连续重复字符 ：

• 如果这一组长度为 1 ，则将字符追加到 s 中。
• 否则，需要向 s 追加字符，后跟这一组的长度。

压缩后得到的字符串 s 不应该直接返回 ，需要转储到字符数组 chars 中。需要注意的是，如果组长度为 10 或 10 以上，则在 chars 数组中会被拆分为多个字符。 请在 修改完输入数组后 ，返回该数组的新长度。 你必须设计并实现一个只使用常量额外空间的算法来解决此问题。 注意： 数组中超出返回长度的字符无关紧要，应予忽略。

示例 1：

输入： chars = "a","a","b","b","c","c","c"输出： 返回 6 ，输入数组的前 6 个字符应该是："a","2","b","2","c","3"解释："aa" 被 "a2" 替代。"bb" 被 "b2" 替代。"ccc" 被 "c3" 替代。

示例 2：

输入： chars = "a"输出： 返回 1 ，输入数组的前 1 个字符应该是："a"解释： 唯一的组是“a”，它保持未压缩，因为它是一个字符。

示例 3：

输入： chars = "a","b","b","b","b","b","b","b","b","b","b","b","b"输出： 返回 4 ，输入数组的前 4 个字符应该是："a","b","1","2"。 解释： 由于字符 "a" 不重复，所以不会被压缩。"bbbbbbbbbbbb" 被 “b12” 替代。

提示：

• 1 <= chars.length <= 2000
• chars[i] 可以是小写英文字母、大写英文字母、数字或符号

参考解法

下面保留的是原始练习仓库中的个人作答，可在右侧工作台中独立重写，再按需揭示对照。

class Solution(object):
    def compress(self, chars):
        if not chars:
            return 0

        write = read = 0
        n = len(chars)

        while read < n:
            current_char = chars[read]  # 当前遍历字符
            count = 0  # 统计出现次数
            while read < n and chars[read] == current_char:
                count += 1
                read += 1
            chars[write] = current_char  # 写入当前字符
            write += 1
        
            if count > 1:
                count_str = str(count)  # 数字转换字符串
                for digit in count_str:
                    chars[write] = digit
                    write += 1
        
        return write
        """
        :type chars: List[str]
        :rtype: int
        """