String Compression

[tom4649]

https://leetcode.com/problems/string-compression/description/

[nodchip]

str(right - left) で、O(log chars.length) の文字数のメモリが確保されると思うのですが、それが constant extra space と言えるのかどうか気になりました。かりに str をしようしないとしたら、どのようなロジックにすればよいでしょうか？

[tom4649]

ご指摘の通りで、count の桁数分 O(log chars.length) の空間計算量になりますね。自力で気付けませんでした。以下の関数を定義することで constant extra space になるかと思います。

def append_count(length_before, count):
        length_after = length_before
        while count > 0:
            chars[length_after] = str(count % 10)
            count //= 10
            length_after += 1

        left = length_before
        right = length_after - 1
        while left < right:
            chars[left], chars[right] = chars[right], chars[left]
            left += 1
            right -= 1

        return length_after