[이코테/프로그래머스/구현] 문자열 압축

문제

데이터 처리 전문가가 되고 싶은 "어피치"는 문자열을 압축하는 방법에 대해 공부를 하고 있습니다. 최근에 대량의 데이터 처리를 위한 간단한 비손실 압축 방법에 대해 공부를 하고 있는데, 문자열에서 같은 값이 연속해서 나타나는 것을 그 문자의 개수와 반복되는 값으로 표현하여 더 짧은 문자열로 줄여서 표현하는 알고리즘을 공부하고 있습니다.간단한 예로 "aabbaccc"의 경우 "2a2ba3c"(문자가 반복되지 않아 한번만 나타난 경우 1은 생략함)와 같이 표현할 수 있는데, 이러한 방식은 반복되는 문자가 적은 경우 압축률이 낮다는 단점이 있습니다. 예를 들면, "abcabcdede"와 같은 문자열은 전혀 압축되지 않습니다. "어피치"는 이러한 단점을 해결하기 위해 문자열을 1개 이상의 단위로 잘라서 압축하여 더 짧은 문자열로 표현할 수 있는지 방법을 찾아보려고 합니다.
예를 들어, "ababcdcdababcdcd"의 경우 문자를 1개 단위로 자르면 전혀 압축되지 않지만, 2개 단위로 잘라서 압축한다면 "2ab2cd2ab2cd"로 표현할 수 있습니다. 다른 방법으로 8개 단위로 잘라서 압축한다면 "2ababcdcd"로 표현할 수 있으며, 이때가 가장 짧게 압축하여 표현할 수 있는 방법입니다.
다른 예로, "abcabcdede"와 같은 경우, 문자를 2개 단위로 잘라서 압축하면 "abcabc2de"가 되지만, 3개 단위로 자른다면 "2abcdede"가 되어 3개 단위가 가장 짧은 압축 방법이 됩니다. 이때 3개 단위로 자르고 마지막에 남는 문자열은 그대로 붙여주면 됩니다.
압축할 문자열 s가 매개변수로 주어질 때, 위에 설명한 방법으로 1개 이상 단위로 문자열을 잘라 압축하여 표현한 문자열 중 가장 짧은 것의 길이를 return 하도록 solution 함수를 완성해주세요

정답

def solution(s): 
    answer = len(s)
    for step in range(1,len(s)//2+1):     
        prev = s[:step]
        cnt=1
        com = ""
        for i in range(step,len(s),step):
            # 이전의 값과 똑같다면 cnt+=1
            if prev == s[i:i+step]:
                cnt+=1 
            # 새로운 값이 나왔는데 cnt가 2 이상이면 압축문자열 com 갱신 + prev 재설정
            # 새로운 값이 나왔는데 cnt가 2 미만이면 prev 재설정만
            else:
                # if-else 조건표현식
                # com = 반복횟수 + 반복문자열 + 나머지
                com += str(cnt)+prev  if cnt >=2 else prev
                prev = s[i:i+step] 
                cnt = 1
                   
            
        com+=str(cnt)+prev if cnt>=2 else prev
        answer = min(answer,len(com))
                    
    
    return answer

풀이

문제를 다 읽고 나서 반복되는 단위의 최대값은 문자열 길이/2 일 것임을 유추했고

1부터 (문자열 길이/2) 까지 완전탐색으로풀면 되겠다는 생각은 했다

근데 생각한 로직을 코드로 구현하기가 어려웠음(그래서 구현문제겠지..)

 

정답을 참고해서 풀었다

문제풀이 과정은 다음과 같음

 

1. 문자열을 나눌 단위를 step으로 잡고, step은 1부터 len(s)//2 + 1까지 1씩 증가시켜가며 반복문을 돌림

 

2. 처음에 문자열 s의 시작부터 step까지를 prev 변수에 담아 비교할 문자열의 원본을 생성

 

3. 문자열s의 길이만큼 반복문을 돌면서 prev와 prev의 바로 다음 구간을 비교

(예를 들어, 문자열 abcdcc가 있고 step이 2일 때, 처음 prev값은 'ab'이고 (s[:2])

s[2:4] 의 값 'cd'와 같은지 비교)

 

4. 만약 두 값이 같다면 cnt를 1 증가시킴. 여기서 cnt는 반복횟수를 나타내는 값으로

처음 prev를 설정할 때 이미 그 값이 1번 포함되었다고 치기 때문에 cnt = 1로 초기화한다

 

5. 두 값이 다르다면(이전과 다른 새로운 문자열이 등장했다면) 

prev를 새로운 문자열로 갱신시켜야 함(이제 이 prev값을 다음번 나올 문자열과 일치하는지 비교)

 

5-1. 이 때, 만약 cnt가 2 이상이라면 (반복이 발생했으면) com이라는 문자열변수에 지금까지의 결과를 압축해서 저장

 

문제의 조건에 맞게 반복되는 횟수(cnt) +  반복되는 문자열인 prev 를 붙임

5-2 하지만 cnt가 1이라면 반복이 일어나지 않은 것으로 com변수에 그냥 prev만 추가

 

6. for문이 끝나고 마지막에 문자가 남았다면 5-1,5-2과정을 참고해서 반복이면 압축해서, 반복이 아니라면 그냥 나머지를 com에 붙여 문자열을 완성해준다

 

7.  이 과정을 step을 증가시켜가며 반복하고,

그때마다 압축시킨 최종 문자열의 길이는 이전과 비교해서 더 작은 값으로 갱신한다