压缩字符串

  • 题目描述:给定一组字符,使用原地算法将其压缩。

    压缩后的长度必须始终小于或等于原数组长度。

    数组的每个元素应该是长度为1 的字符(不是 int 整数类型)。

    在完成原地修改输入数组后,返回数组的新长度。

    进阶:
    你能否仅使用O(1) 空间解决问题?

    示例 1:

    输入:
    ["a","a","b","b","c","c","c"]

    输出:
    返回6,输入数组的前6个字符应该是:["a","2","b","2","c","3"]
    说明:
    "aa"被"a2"替代。"bb"被"b2"替代。"ccc"被"c3"替代。

    示例 2:
    输入:
    ["a"]

    输出:
    返回1,输入数组的前1个字符应该是:["a"]

    说明:
    没有任何字符串被替代。

    示例 3:
    输入:
    ["a","b","b","b","b","b","b","b","b","b","b","b","b"]

    输出:
    返回4,输入数组的前4个字符应该是:["a","b","1","2"]。

    说明:
    由于字符"a"不重复,所以不会被压缩。"bbbbbbbbbbbb"被“b12”替代。
    注意每个数字在数组中都有它自己的位置。
    注意:
    所有字符都有一个ASCII值在[35, 126]区间内。
    1 <= len(chars) <= 1000。

  • 链接:https://leetcode-cn.com/problems/string-compression


  • 解题思路:
  1. 使用读写双指针 read & write
  2. 添加 anchor 指针 代表连续字符串的起始位置
  3. 当read指针 + 1 等于字符串长度 或 read指针 + 1不指向同一字符时,开始write
    4.write指针写入内容包括当前anchor指针指向的字符,以及(read - anchor + 1)的字符长度
  • Python版:
class Solution(object):
    def compress(self, chars):
        anchor = write = 0
        for read, c in enumerate(chars):
            if read + 1 == len(chars) or chars[read + 1] != c:
                chars[write] = chars[anchor]
                write += 1
                if read > anchor:
                    for digit in str(read - anchor + 1):
                        chars[write] = digit
                        write += 1
                anchor = read + 1
        return write

  • Tips:每次遍历结束后,anchor指针需重新指向新的连续字符起始处

  • 复杂度分析

    • 时间复杂度:O(N)O(N),其中 NN 是 chars 的长度。
    • 空间复杂度:O(1)O(1),三个指针的占用空间。
  • Java版:

class Solution {
    public int compress(char[] chars) {
        int anchor = 0, write = 0;
        for (int read = 0; read < chars.length; read++) {
            if (read + 1 == chars.length || chars[read + 1] != chars[read]) {
                chars[write++] = chars[anchor];
                if (read > anchor) {
                    for (char c: ("" + (read - anchor + 1)).toCharArray()) {
                        chars[write++] = c;
                    }
                }
                anchor = read + 1;
            }
        }
        return write;
    }
}

©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容