leetcode_443 压缩字符串

120 阅读2分钟

要求

给你一个字符数组 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"]
解释:
没有任何字符串被替代。

示例 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:
    def compress(self, chars: List[str]) -> int:
        count = 1
        length = len(chars)

        for index in range(length - 1,-1,-1):
            if index > 0 and chars[index] == chars[index - 1]:
                count += 1
            else:
                end = index + count
                if count == 1:
                    chars[index:end] = [chars[index]]
                else:
                    chars[index:end] = [chars[index]] + list(str(count))
                    count = 1
        return len(chars)

image.png

解题思路:我们从后向前遍历数据列表,从后向前遍历,这样在压缩的时候就不用更新索引,然后我们遇到相同的就进行计数增加,然后遇到不同的字符的时候进行替换,注意下列表换值的技巧,比较简单。

>>> a = [1,2,3,4,5,6]
>>> a[2:4] = [0]
>>> a
[1, 2, 0, 5, 6]
>>>