leetcode 443. 压缩字符串

leetcode 443. 压缩字符串

一、题目

原题链接：443. 压缩字符串

1.题目描述

给你一个字符数组 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] 可以是小写英文字母、大写英文字母、数字或符号

2.基础框架

C++基础框架代码如下:

int compress(vector<char>& chars) {

3.源码实现

int compress(vector<char>& chars) {
    if (chars.size() == 1) return 1;
    int pos = 0;	
    int i, j;	// 快慢指针
    for (i = 0, j = 1; j < chars.size(); ++j)
    {
        if (chars[i] == chars[j])
        {
            continue;
        }
        else if (j - i == 1)	// i和j不同，且i和j的下标只相差1
        {
            chars[pos] = chars[i];
        }
        else	// i和j对应的字母不同，且i和j的下标相差大于1，说明[i,j)中有多个相同的字母，存在需要压缩成数字的情况
        {
            int cnt = j - i;	// j和i的下标差：有多少个连续相同的字母数量
            stack<int> st;	// 如果12个相同的字母，字母后面要追加‘1’,‘2’,利用堆栈方便在于可以先取到高位
            chars[pos] = chars[i];	// 将相同的字母先存入pos下标的位置，之后就可以放数字了
            while (cnt)
            {
                st.push(cnt % 10);
                cnt /= 10;
            }
            while (!st.empty())
            {
                chars[++pos] = st.top() + '0';	// 先将高位取出
                st.pop();
            }
        }
        ++pos;
        i = j;
    }
    chars[pos] = chars[i];
    if (j - 1 != i && chars[j - 1] == chars[i])
    {
        int cnt = j - i;
        stack<int> st;
        while (cnt)
        {
            st.push(cnt % 10);
            cnt /= 10;
        }
        while (!st.empty())
        {
            chars[++pos] = st.top() + '0';
            st.pop();
        }
    }
    return pos + 1;
}