压缩字符串 简单 Python

1. 题目
   给定一组字符，使用原地算法将其压缩。
   压缩后的长度必须始终小于或等于原数组长度。
   数组的每个元素应该是长度为1 的字符（不是 int 整数类型）。
   在完成原地修改输入数组后，返回数组的新长度。
   进阶：
   你能否仅使用O(1) 空间解决问题？
   例子：
   [“a”,“a”,“b”,“b”,“c”,“c”,“c”]
   输出：
   返回6，输入数组的前6个字符应该是：[“a”,“2”,“b”,“2”,“c”,“3”]
   说明：
   "aa"被"a2"替代。"bb"被"b2"替代。"ccc"被"c3"替代。
   来源：力扣（LeetCode）
   链接：https://leetcode-cn.com/problems/string-compression
   著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

2. 思路

   • 题目没有说给的是连续的，当时自己想多了，多看看例子总是好的（狗头）
   • 要原地修改数组，先看需要的变量：遍历数组需要一个read指针；如果字符出现超过两次就要进行写操作，因此需要一个write指针；写指针一般会小于read指针，但是还要记录新的字符出现次数，需要一个记录新字符起始位置的指针。
   • 空数组和长度为1的数组返回0和1。

3. 代码

    class Solution:
    def compress(self, chars: List[str]) -> int:
        if not chars or len(chars) == 1:
            return len(chars)
        anchor = 0 # 记录第一个与前面字符不同的字符的下标
        write = 0 # 记录返回数组的最后一个值的下标
        for read, ch in enumerate(chars): # read记录当前在读的位置
            if read + 1 == len(chars) or chars[read+1] != ch:
                chars[write] = chars[anchor]
                write += 1
                if read > anchor:
                    for item in str(read - anchor + 1):
                        chars[write] = item
                        write += 1
                anchor = read + 1
            else:
                continue
        return write

4. 知识点
   • enumerate的用法（遍历可迭代的对象）：key-value。
   • 判断是否到数组尾端或者新的字符的语句的写法，不能交换位置！！！
   • 多指针配合使用。