0443-string-compression.py

# 443. String Compression
# https://leetcode.com/problems/string-compression

classSolution:
defcompress(self, chars: list[str]) ->int:
n=len(chars)

ifn==1:
return1

i=0
j=0
count=0

whilei<n:
count+=1

ifi==n-1orchars[i+1] !=chars[i]:
chars[j] =chars[i]
j+=1

ifcount>1:
forcharinstr(count):
chars[j] =char
j+=1

count=0

i+=1

returnj

# ********************#
# TEST #
# ********************#

importunittest

classTestStringMethods(unittest.TestCase):
deftest_addBinary(self):
self.assertEqual(Solution.compress(self, ["a", "a", "b", "b", "c", "c", "c"]), 6)
self.assertEqual(Solution.compress(self, ["a"]), 1)
self.assertEqual(Solution.compress(self, ["a", "b", "b", "b", "b", "b", "b", "b", "b", "b", "b", "b", "b"]), 4)

if__name__=='__main__':
unittest.main()