0443-string-compression.rb

# frozen_string_literal: true

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

# @param {Character[]} chars
# @return {Integer}
defcompress(chars)
n=chars.length

return1ifn == 1

i=0
j=0
count=0

whilei < ndo
count += 1

ifchars[i + 1] != chars[i]
chars[j]=chars[i]
j += 1

ifcount > 1
count.to_s.each_chardo |char|
chars[j]=char
j += 1
end
end

count=0
end

i += 1
end

j
end

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

require"test/unit"
classTest_compress < Test::Unit::TestCase
deftest_
assert_equal6,compress(["a","a","b","b","c","c","c"])
assert_equal1,compress(["a"])
assert_equal4,compress(["a","b","b","b","b","b","b","b","b","b","b","b","b"])
end
end