[Python3][2020카카오공채] 문자열 압축

문제 :

https://www.welcomekakao.com/learn/courses/30/lessons/60057

나의 풀이 :

def solution(s):

    def compress(size):

        ret = 0

        before, count = s[:size], 1

        for i in range(size, len(s), size):

            word = s[i:i+size]

            if word == before:

                count += 1

            else:

                if count > 1:

                    ret += len(str(count))

                ret += size

                before, count = word, 1

        if count > 1:

            ret += len(str(count))

        ret += len(before)

        return ret

    ans = len(s)

    for size in range(1, ans+2//2):

        ans = min(ans, compress(size))

    return ans