【每日一题】力扣443.压缩字符串

题目描述（传送门）

给定一组字符，使用原地算法将其压缩。

压缩后的长度必须始终小于或等于原数组长度。

数组的每个元素应该是长度为1 的字符（不是 int 整数类型）。

在完成原地修改输入数组后，返回数组的新长度。

示例 1：

输入：
["a","a","b","b","c","c","c"]

输出：
返回 6 ，输入数组的前 6 个字符应该是：["a","2","b","2","c","3"]

说明：
"aa" 被 "a2" 替代。"bb" 被 "b2" 替代。"ccc" 被 "c3" 替代。

示例 2：

输入：
["a"]

输出：
返回 1 ，输入数组的前 1 个字符应该是：["a"]

解释：
没有任何字符串被替代。

示例 3：

输入：
["a","b","b","b","b","b","b","b","b","b","b","b","b"]

输出：
返回 4 ，输入数组的前4个字符应该是：["a","b","1","2"]。

解释：
由于字符 "a" 不重复，所以不会被压缩。"bbbbbbbbbbbb" 被 “b12” 替代。
注意每个数字在数组中都有它自己的位置。

解题思路&代码解析

import java.util.Arrays;

/**
 * @ClassName Main
 * @Description ：TODO
 * @Author Josvin
 * @Date 2021/05/13/10:49
 */
public class Main {
    public  static int compress(char[] chars) {
       // 原地算法
        /*
        *
        * 输入：
["a","a","b","b","c","c","c"]


输出：
返回 6 ，输入数组的前 6 个字符应该是：["a","2","b","2","c","3"]

说明：
"aa" 被 "a2" 替代。"bb" 被 "b2" 替代。"ccc" 被 "c3" 替代。

         *  */

        // index  相当于一个更新指针
        int index = 0;
        int count = 1;int i = 1;
        for (; i < chars.length; i++) {
            if (chars[i] == chars[i - 1]) {
                count++;
            } else {
                if (count == 1) {
                    chars[index] = chars[i - 1];
                    index++;
                    continue;
                }
                chars[index++] = chars[i - 1];
                //  考虑 count 大于10的情况
                int left = index;
                while (count > 0) {
                    chars[index++] = (char) ('0' + count % 10);
                    count = count / 10;
                }
                // 因为这样处理出来的结果，是先存的低位后存储高位，这样刚好是反的，进行反转
                int right = index - 1;
                while (left < right) {
                    char tmp = chars[left];
                    chars[left] = chars[right];
                    chars[right] = tmp;
                    left++;
                    right--;
                }
                count = 1;
            }
        }

        if (count == 1) {
            chars[index++] = chars[i - 1];
        } else {
            chars[index++] = chars[i - 1];
            int left = index;
            while (count>0) {
                chars[index++] = (char) ('0' + count % 10);
                count = count / 10;
            }
            int right = index - 1;
            while (left < right) {
                char tmp = chars[left];
                chars[left] = chars[right];
                chars[right] = tmp;
                left++;
                right--;
            }
        }
        return index;
    }

    public static void main(String[] args) {
        char[] chars = {'b','b','b','b','b','b','b','b','b','b'};
        System.out.println(compress(chars));
    }
}

发现解决了去年的遗留问题。

上边代码比较冗余，自己也可以进行优化，自己看了些题解，再给出简练的代码：

方法二

/**
 * @ClassName Main2
 * @Description ：TODO
 * @Author Josvin
 * @Date 2021/05/13/20:34
 */
public class Main2 {
    public static int compress(char[] chars) {
        int index = 0;
        int i = 0;
        while (i < chars.length) {
            int beg = i;
            while (i < chars.length && chars[i] == chars[beg]) {
                i++;
            }
            int count = i - beg;
            chars[index++] = chars[beg];
            // count 》1 进行存放，需要考虑大于9的情况
            if (count > 1) {
                for (char c:Integer.toString(count).toCharArray()) {
                    chars[index++] = c;
                }
            }
        }
        return index;
    }
    public static void main(String[] args) {
        char[] chars = {'b','b','b','b','b','b','b','b','b','b'};
        System.out.println(compress(chars));
    }
}