Question

443. 压缩字符串

English Version

题目描述

给你一个字符数组 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"]
解释：唯一的组是“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:
    i, k, n = 0, 0, len(chars)
    while i < n:
      j = i + 1
      while j < n and chars[j] == chars[i]:
        j += 1
      chars[k] = chars[i]
      k += 1
      if j - i > 1:
        cnt = str(j - i)
        for c in cnt:
          chars[k] = c
          k += 1
      i = j
    return k

class Solution {
  public int compress(char[] chars) {
    int k = 0, n = chars.length;
    for (int i = 0, j = i + 1; i < n;) {
      while (j < n && chars[j] == chars[i]) {
        ++j;
      }
      chars[k++] = chars[i];
      if (j - i > 1) {
        String cnt = String.valueOf(j - i);
        for (char c : cnt.toCharArray()) {
          chars[k++] = c;
        }
      }
      i = j;
    }
    return k;
  }
}

class Solution {
public:
  int compress(vector<char>& chars) {
    int k = 0, n = chars.size();
    for (int i = 0, j = i + 1; i < n;) {
      while (j < n && chars[j] == chars[i])
        ++j;
      chars[k++] = chars[i];
      if (j - i > 1) {
        for (char c : to_string(j - i)) {
          chars[k++] = c;
        }
      }
      i = j;
    }
    return k;
  }
};

func compress(chars []byte) int {
  i, k, n := 0, 0, len(chars)
  for i < n {
    j := i + 1
    for j < n && chars[j] == chars[i] {
      j++
    }
    chars[k] = chars[i]
    k++
    if j-i > 1 {
      cnt := strconv.Itoa(j - i)
      for _, c := range cnt {
        chars[k] = byte(c)
        k++
      }
    }
    i = j
  }
  return k
}

impl Solution {
  pub fn compress(chars: &mut Vec<char>) -> i32 {
    let (mut i, mut k, n) = (0, 0, chars.len());
    while i < n {
      let mut j = i + 1;
      while j < n && chars[j] == chars[i] {
        j += 1;
      }
      chars[k] = chars[i];
      k += 1;

      if j - i > 1 {
        let cnt = (j - i).to_string();
        for c in cnt.chars() {
          chars[k] = c;
          k += 1;
        }
      }
      i = j;
    }
    k as i32
  }
}