【Lintcode】213. String Compression

题目地址：

https://www.lintcode.com/problem/string-compression/description

给定一个字符串$s$，要求其按照如下规则压缩：某个字母$c$如果连续出现了$k$次，那么就记为$ck$。如果字符串压缩后长度变小，则返回压缩后的字符串，否则返回原字符串。

思路是，每次遍历到一个字母的时候就数一下它出现了多少次，然后append到答案后面即可。代码如下：

public class Solution {
    /**
     * @param originalString: a string
     * @return: a compressed string
     */
    public String compress(String originalString) {
        // write your code here
        StringBuilder sb = new StringBuilder();
        int count;
        for (int i = 0; i < originalString.length(); i++) {
            count = 1;
            while (i + 1 < originalString.length() && originalString.charAt(i) == originalString.charAt(i + 1)) {
                i++;
                count++;
            }
            sb.append(originalString.charAt(i)).append(count);
        }
        
        return sb.length() < originalString.length() ? sb.toString() : originalString;
    }
}

时空复杂度$O(n)$。