[프로그래머스/Programmers] 문자열 압축 - Python

Link

코딩테스트 연습 - 문자열 압축

 

Solution

def solution(s):
    lens = []

    if len(s) == 1:
        return 1

    compact = ''
    for i in range(1, len(s) // 2 + 1):
        count = 1
        t = s[:i]
        for j in range(i, len(s), i):
            if s[j:j + i] == t:
                count += 1
            else:
                if count == 1:
                    count = ''
                compact += str(count) + t
                t = s[j: j + i]
                count = 1

        if count == 1:
            count = ''
        compact += str(count) + t
        lens.append(len(compact))
        compact = ''

    return min(lens)

🔑 key point:
        for j in range(i, len(s), i):
            if s[j:j + i] == t:
                count += 1
            else:
                if count == 1:
                    count = ''
                compact += str(count) + t
                t = s[j: j + i]
                count = 1

 

Other Solution

def compress(text, tok_len):
    words = [text[i:i + tok_len] for i in range(0, len(text), tok_len)]
    res = []
    cur_word = words[0]
    cur_cnt = 1
    for a, b in zip(words, words[1:] + ['']):
        if a == b:
            cur_cnt += 1
        else:
            res.append([cur_word, cur_cnt])
            cur_word = b
            cur_cnt = 1
    return sum(len(word) + (len(str(cnt)) if cnt > 1 else 0) for word, cnt in res)

def solution(text):
    return min(compress(text, tok_len) for tok_len in list(range(1, len(text) // 2 + 1)) + [len(text)])