题目描述

给你一个字符数组 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" 替代。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/string-compression
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

思路分析

• 今天的每日一题，依旧是字符串问题，根据题意，压缩算法已经给出，我们需要根据压缩算法实现代码。

• 首先，使用朴素解法，将字符串完成压缩，使用StringBuilder 提升效率。压缩完成之后，在对原数组进行复写，返回结果。

• 但是题目要求只使用常量额外空间，因此，朴素解法不符合要求。采用双指针的方式优化代码。

代码

• 朴素解法

    public int compress(char[] chars) {
        StringBuilder builder = new StringBuilder();
        int n = chars.length;
        char temp = chars[0];
        int cnt = 0;
        for (int i = 0; i < n; i++) {
            if (temp == chars[i]) {
                cnt++;
            } else {
                builder.append(temp);
                if (cnt != 1) {
                    builder.append(cnt);
                }
                temp = chars[i];
                cnt = 1;
            }
        }
        
        builder.append(temp);
        if (cnt != 1) {
            builder.append(cnt);
        }

        String s = builder.toString();
        int ans = s.length();
        for (int i = 0 ; i < ans; i++) {
            chars[i] = s.charAt(i);
        }
        return ans;
    }

• 双指针解法

    public int compress(char[] chars) {
        int n = chars.length;
        int i = 0; 
        int j = 0;
        while (i < n) {
            int idx = i;
            while (idx < n && chars[idx] == chars[i]) {
                idx++;
            }
            int cnt = idx - i;
            chars[j++] = chars[i];
            if (cnt > 1) {
                int start = j, end = start;
                while (cnt != 0) {
                    chars[end++] = (char) ((cnt % 10) + '0');
                    cnt /= 10;
                }
                reverse(chars, start, end - 1);
                j = end;
            }
            i = idx;
        }

        return j;
    }

    public void reverse(char[] cs, int start, int end) {
        while (start < end) {
            char t = cs[start];
            cs[start] = cs[end];
            cs[end] = t;
            start++; end--;
        }
    }

总结

• 朴素解法时间复杂度是O(n),空间复杂度是O(n)

• 双指针解法时间复杂度是O(n),空间复杂度是O(1)

• 坚持算法每日一题，加油！