字符串压缩

题目描述

来源于 https://leetcode-cn.com/

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

示例1:

 输入:"aabcccccaaa"
 输出:"a2b1c5a3"

示例2:

 输入:"abbccd"
 输出:"abbccd"
 解释:"abbccd"压缩后为"a1b2c2d1",比原字符串长度更长。

提示:

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

解法:

class Solution {
public:
    string compressString(string s) {
        // 无须压缩
        if(s.size() <= 1){
            return s;
        }

        ostringstream os;
        os << s[0];
        int n = 1;
        for(size_t i=1;i<s.size();i++){
            if(s[i] == s[i-1]){
                n += 1;
            }else{
                os << n << s[i];
                n = 1;
            }
        }
        os << n;
        
        string result = os.str();

        if(result.size() < s.size()){
            return result;
        }else{
            return s;
        }
    }
};