【算法】压缩字符串

90 阅读1分钟

难度:中等

题目:

给你一个字符数组 chars ,请使用下述算法压缩:

从一个空字符串 s 开始。对于 chars 中的每组 连续重复字符

  • 如果这一组长度为 1 ,则将字符追加到 s 中。
  • 否则,需要向 s 追加字符,后跟这一组的长度。

压缩后得到的字符串 s 不应该直接返回 ,需要转储到字符数组 chars 中。需要注意的是,如果组长度为 1010 以上,则在 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"]
解释:唯一的组是“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] 可以是小写英文字母、大写英文字母、数字或符号

解题思路:

  1. 双指针法:使用两个指针 readwriteread 指针用于读取原始数据,而 write 指针用于写入压缩后的数据。
  2. 计数器:使用一个计数器 count 来记录连续字符的数量。
  3. 循环遍历:遍历整个数组,当发现当前字符与前一个字符不同时,更新 write 指针的位置,并写入当前字符及其出现次数(如果次数大于 1)。
  4. 更新位置:每次处理完一个字符后,更新 read 指针,开始处理下一个字符。
  5. 写入字符:当 count 大于 1 时,需要写入数字。由于数字可能为多位数,需要逐个写入。
  6. 返回新长度:最后返回 write 指针的位置作为新数组的长度。

JavaScript实现:

/**
 * @param {character[]} chars
 * @return {number}
 */
var compress = function(chars) {
  let read = 0; // 读取指针
  let write = 0; // 写入指针
  let count = 1; // 计数器
  while (read < chars.length) {
    // 当前字符与下一个字符相等时,增加计数器
    if (read + 1 < chars.length && chars[read] === chars[read + 1]) {
      count++;
      read++;
    } else {
      // 写入当前字符
      chars[write] = chars[read];
      write++;

      // 如果计数器大于 1,则写入计数器
      if (count > 1) {
        const strCount = count.toString();
        for (let i = 0; i < strCount.length; i++) {
          chars[write] = strCount[i];
          write++;
        }
      }


      // 更新计数器并移动读取指针
      count = 1;
      read++;
    }
  }

  // 返回新长度
  return write;
}