面试题 01.06. 字符串压缩

[yankewei]

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

示例1:

 输入："aabcccccaaa"
 输出："a2b1c5a3"

示例2:

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

提示：

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/compress-string-lcci

解答

没有什么可说的，水题

func compressString(S string) string {
    var t strings.Builder
    var cnt int
    var end byte
    for i := 0; i < len(S); i++ {
	lenT := t.Len()
	if lenT == 0 { // t为空
	    t.WriteByte(S[i])
	    end = S[i]
	    cnt = 1
	} else {
	    if S[i] == end { // t的最后一个字符和当前遍历的字符一样
		cnt++
	    } else {
		t.WriteString(strconv.Itoa(cnt))
		end = S[i]
		t.WriteByte(end)
		cnt = 1
	    }
	}
    }
    t.WriteString(strconv.Itoa(cnt))
    if len(S) > t.Len() {
	return t.String()
    }
    return S
}