刷题的日常-压缩字符串

90 阅读2分钟

携手创作,共同成长!这是我参与「掘金日新计划 · 8 月更文挑战」的第15天,点击查看活动详情

刷题的日常

一天一题,保持脑子清爽

压缩字符串

来自leetcode的 443 题,题意如下:

给你一个字符数组 chars ,请使用下述算法压缩:
从一个空字符串 s 开始。对于 chars 中的每组 连续重复字符 :

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

压缩后得到的字符串 s 不应该直接返回 ,需要转储到字符数组 chars 中。需要注意的是,如果数组长度为 10 或 10 以上,则在 chars 数组中会被拆分为多个字符。
请在 修改完输入数组后 ,返回该数组的新长度。
你必须设计并实现一个只使用常量额外空间的算法来解决此问题。

理解题意

我们可以从题意中提取的条件如下:

  • 题目给出一个字符数组
  • 我们需要将数组进行压缩
    • 如果一个字符连续出现,先将字符接上去
    • 统计出现的次数并把次数放在字符后面
  • 只能在原地进行操作
  • 最后返回压缩后的字符长度

做题思路

我们可以开辟额外空间进行操作,然后将数据拷贝回原数组。这里我们不开辟空间,直接在原数组上操作:

  • 遍历数组,统计同一个字符连续出现的次数
  • 如果出现不同的字符,将字符和次数写回数组
  • 最后如果统计的次数不为0,则需要进行再次写入

代码实现

代码实现如下:

class Solution {
    public int compress(char[] chars) {
        int len = 0;
        int count = 0;
        char temp = chars[0];
        for (int i = 0; i < chars.length; i++) {
            char data = chars[i];
            if (data == temp) {
                //相等
                count++;
                continue;
            }
            //不相等
            chars[len] = temp;
            if (count > 1) {
                String s = count + "";
                char[] nums = s.toCharArray();
                for (char c : nums) {
                    chars[++len] = c;
                }
            }
            len++;
            //重置变量
            temp = data;
            count = 1;
        }
        chars[len] = temp;
        if (count > 1) {
            chars[len] = chars[chars.length - 1];
            String s = count + "";
            char[] nums = s.toCharArray();
            for (char c : nums) {
                chars[++len] = c;
            }
        }
        return ++len;
    }
}