[프로그래머스] 문자열 압축

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

다른사람 풀이 (성공)

def solution(s):
    answer = len(s)
    # 1개 단위(step)부터 압축 단위를 늘려간다.
    for step in range(1, len(s) // 2 + 1):  
        compressed = ''
        prev = s[:step] # 앞에서부터 step만큼의 문자열 출력
        count = 1
        
        # 단위 크기 만큼 증가시키며, 이전 문자열과 비교
        for j in range(step, len(s), step):
            
            # 이전 횟수와 동일하면, 압축 횟수(count) 증가
            if prev == s[j:j + step]:
                count += 1
            
            # 다른 문자열이 나오면
            else:
                compressed += str(count) + prev if count >= 2 else prev
                prev = s[j : j + step] # 다시 초기화
                count = 1
        
        # 남아 있는 문자열에 대해서 처리
        compressed += str(count) + prev if count >= 2 else prev
        
        # 만들어지는 문자열이 가장 짧은 것이 정답
        answer = min(answer, len(compressed))
            
            
    
    return answer

다른방법 1

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, int(len(text)/2) + 1)) + [len(text)])

a = [
    "aabbaccc",
    "ababcdcdababcdcd",
    "abcabcdede",
    "abcabcabcabcdededededede",
    "xababcdcdababcdcd",

    'aaaaaa',
]

for x in a:
    print(solution(x))