面试题01.06_字符串压缩_hn

2020-03-16  本文已影响0人  1只特立独行的猪

题目描述

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

示例

示例 1:

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

示例2:

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

解答方法

方法一:遍历

思路

从左到右遍历字符串,用ch记录当前要压缩的字符,cnt记录当前字符的次数,如果当前字符S[i]等于ch,cnt = cnt + 1,否则将ch以及cnt更新到结果字符串res中,即res = res+ch+cnt, 完成对字符ch的压缩随后更新ch为s[i],cnt为1。
在遍历结束之后,我们就得到了压缩后的字符串 res,并将其长度与原串长度进行比较。如果长度没有变短,则返回原串,否则返回压缩后的字符串。

代码

class Solution:
    def compressString(self, S: str) -> str:
        if not S:
            return ''
        res = ''
        ch = S[0]
        cnt = 0
        for s in S:
            if s == ch:
                cnt += 1
            else:
                res += ch + str(cnt)
                ch = s
                cnt = 1
        res += ch +str(cnt)
        return res if len(res)< len(S) else S

时间复杂度

O(n),其中 n 为字符串的长度,即遍历一次字符串的复杂度。

空间复杂度

O(1),只需要常数空间(不包括存储答案res的空间)存储变量。

上一篇下一篇

猜你喜欢

热点阅读