leetcode 面试题 01.06 字符串压缩 - 难度简单

1. 题目描述

字符串压缩。利用字符重复出现的次数，编写一种方法，实现基本的字符串压缩功能。比如，字符串aabcccccaaa会变为a2b1c5a3。若“压缩”后的字符串没有变短，则返回原先的字符串。你可以假设字符串中只包含大小写英文字母（a至z）。

示例1:

输入：“aabcccccaaa”
输出：“a2b1c5a3”

示例2:

输入：“abbccd”
输出：“abbccd”
解释：“abbccd"压缩后为"a1b2c2d1”，比原字符串长度更长。
提示：

字符串长度在[0, 50000]范围内。

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

2. 题解

题解1（可以优化）

我拿到题目以后，想到的思路是设置双指针，遍历字符串，记录字符及出现次数，这样的时间复杂度应该还是O(n)吧？但是出现了while嵌套。

var compressString = function(S) {
    const len = S.length;
    if(len < 2){
        return S;
    }
    let i = 0;
    let j = 0;
    let res = "";
    while(j < len){
        res += S.charAt(i);
        while(S.charAt(i) == S.charAt(j)){
            j++;
        }
        res += j - i;
        i = j;
    }
    if(res.length < S.length){
        return res;
    }else{
        return S;
    }
};

题解2

初始化curr和count, curr用于记录每次待加入结果的字符，每遍历一个字符都和curr进行比较，

• 若和curr相同，则count++；
• 若不同，说明遍历到不同于curr的字符，此时将curr和count放入结果字符串，然后更新curr和count。

/**
 * @param {string} S
 * @return {string}
 */
var compressString = function(S) {
    const len = S.length;
    if(len < 2){
        return S;
    }
    //初始化变量
    let i = 0;
    let count = 0;
    let curr = S.charAt(0);
    let res = "";
    //遍历字符串
    while(i < len){
        if(S.charAt(i) == curr){
            count++;
        }else{
            res += curr;
            res += count;
            count = 1;
            curr = S.charAt(i);
        }
        i++;
    }
    //最后一次没有可以比较的了，手动加上最后一次的结果
    res += curr;
    res += count;
    //更简洁的写法
    return res.length < S.length ? res : S;
};

时间复杂度：O(n),
空间复杂度：O(1)。