版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/sinat_36811967/article/details/86583010
给定一组字符,使用原地算法将其压缩。
压缩后的长度必须始终小于或等于原数组长度。
数组的每个元素应该是长度为1 的字符(不是 int 整数类型)。
在完成原地修改输入数组后,返回数组的新长度。
进阶:
你能否仅使用O(1) 空间解决问题?
把逻辑理顺就很好写了:
class Solution:
def compress(self, chars):
"""
:type chars: List[str]
:rtype: int
"""
res, count = 0, 1
for i, char in enumerate(chars):
if i == 0:
choose = char
continue
if char == choose:
count += 1
else:
chars[res] = choose
if count != 1:
for one in str(count):
res += 1
chars[res] = one
res += 1
choose = char
count = 1
if i == len(chars)-1:
chars[res] = choose
if count != 1:
for one in str(count):
res += 1
chars[res] = one
return res + 1