[LeetCode] 443. 压缩字符串

443. 压缩字符串
给定一组字符，使用原地算法将其压缩。
压缩后的长度必须始终小于或等于原数组长度。
数组的每个元素应该是长度为1 的字符（不是 int 整数类型）。
在完成原地修改输入数组后，返回数组的新长度。
示例:
输入：
["a","a","b","b","c","c","c"]
输出：
返回6，输入数组的前6个字符应该是：["a","2","b","2","c","3"]
说明：
"aa"被"a2"替代。"bb"被"b2"替代。"ccc"被"c3"替代。

class Solution:
    def compress(self, chars):
        n = len(chars)  #字符的长度
        i, count = 0, 1 #结果字符串长度i, 单个字符个数count
        for j in range(1, n+1): #从第二个字符串开始遍历
            if j < n and chars[j] == chars[j-1]:    #如果当前的字符与前一个字符相同
                count += 1  #字符计数+1
            else:
                chars[i] = chars[j-1]   #否则将压缩后的字符改成前一个字符
                i += 1  #压缩后的字符空位+1
                if count > 1:   #如果当前的计数大于1需要记录字符的个数
                    for m in str(count):    #数字转字符
                        chars[i] = m    #记录数字字符
                        i += 1  #压缩字符空位+1
                count = 1   #计数归1
        return i