【Leetcode】443. String Compression

题目地址：

https://leetcode.com/problems/string-compression/

给定一个长$n$的char型数组$A$，原地将其压缩，并返回压缩后的长度。压缩方式是，如果一个字母$c$连续出现了$m$次，则压缩成$[c,m]$，即字母后面跟上重复次数，次数如果是两位数那就要拆成十位个位来写，三位数以上类似；如果一个字母只出现了$1$次则省略出现次数。

遍历$A$，每次求出当前字符出现次数，然后将字符和次数原地记录下来。代码如下：

class Solution {
 public:
  int compress(vector<char>& v) {
    int res = 0;
    for (int i = 0; i < v.size(); i++) {
      int j = i;
      while (j < v.size() && v[i] == v[j]) j++;
      int cnt = j - i;
      v[res++] = v[i];
      if (cnt > 1) {
        auto s = to_string(cnt);
        for (char ch : s) v[res++] = ch;
      }

      i = j - 1;
    }

    return res;
  }
};

时间复杂度$O(n)$，空间$O(1)$。