【leedcode】443. 压缩字符串

259 阅读1分钟

这是我参与8月更文挑战的第22天,活动详情查看:8月更文挑战

难度: 中等

题目描述

给你一个字符数组 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(object):
    def compress(self, chars):
        a, b = 0, 0 
        while b < len(chars):
            if b == len(chars) - 1 or chars[b] != chars[b + 1]:  
                if b != a:  
                    for l in range(len(list(str(b - a + 1)))):  
                        chars.insert(b + l + 1, list(str(b - a + 1))[l])
                    while b != a: 
                        chars.pop(a)
                        b -= 1
                    b += len(list(str(b - a + 1))) + 1  
                    a = b
                else:
                    b += 1
                    a = b
            else:
                b += 1

解法二

队列+栈+集合

在队列chars结尾加上一个结束符'AOF',循环取chars的队首元素,存放在栈 stack。

  • 如果栈中只有一种元素,则循环取队首元素加入栈。
  • 如果栈中出现两种元素,则最后元素出栈,计算当前栈中元素数量len(stack)。len(stack)大于1则在队列chars队尾插入stack[0]+len(stack),len(stack)等于1则在队列chars队尾插入stack[0]。创建新栈时,别忘了加上这个出栈的元素!
  • 遇到结束符,结束程序
class Solution:
    def compress(self, chars: List[str]) -> int:
        stack = []
        chars.append('AOF')
        current = 0
        while current != 'AOF':
            current = chars.pop(0)
            stack.append(current)
            if len(set(stack)) > 1:
                next = stack.pop()
                chars.append(stack[0])
                if len(stack) > 1:
                    chars.extend(list(str(len(stack))))
                stack = [next]
        return len(chars)