题目:
面试题 01.06. 字符串压缩。
字符串压缩。利用字符重复出现的次数,编写一种方法,实现基本的字符串压缩功能。比如,字符串aabcccccaaa会变为a2b1c5a3。若“压缩”后的字符串没有变短,则返回原先的字符串。你可以假设字符串中只包含大小写英文字母(a至z)。
链接:https://leetcode-cn.com/problems/compress-string-lcci
示例:
输入:“aabcccccaaa”
输出:“a2b1c5a3”
输入:“abbccd”
输出:“abbccd”
解释:“abbccd"压缩后为"a1b2c2d1”,比原字符串长度更长。
思路:
利用双指针判断后一个元素是否与当前元素相同。若相同则继续向后比较,若不同则返回与当前元素相同的个数。
在最后输出时,比较压缩字符串与原字符串的长度,选择长度较短的返回。
代码:
class Solution:
def compressString(self, S: str) -> str:
i, j, s = 0, 0, ""
temp = []
while j < len(S):
if S[i] == S[j]:
temp.append(S[i])
j += 1
else:
s += (temp[0]+str(len(temp)))
i = j
temp = []
s += (temp[0]+str(len(temp))) if temp else ''
return s if len(s) < len(S) else S
时间复杂度O(n),空间复杂度O(n).