[프로그래머스] 60057 문자열 압축 (python 파이썬)

https://school.programmers.co.kr/learn/courses/30/lessons/60057

프로그래머스

---

문제에서 문자열은 제일 앞부터 정해진 길이만큼 잘라야 합니다.라는 조건 덕분에 쉽게 해결할 수 있었다.

---

아이디어

1. 문자열 일정 단위로 자르기

• 파이썬에 있는 리스트 슬라이싱을 활용하면 쉽게 해결 가능.
• 자르는 단위를 문자열 길이의 절반 이상이 되면 반복되는 것이 불가능하다. 따라서 자르는 단위는 주어진 문자열 길이의 절반까지 설정한다.

2. 자른 단위에 맞춰 문자열 압축

• for문을 활용하여 반복되는 부분이 있는지 확인한다.

---

전체 코드

def solution(s):

    answer = len(s)
    for i in range(1, len(s) // 2 + 1):
        s_list = []
        for j in range(len(s) // i + 1):
            if s[i * j: i * (j + 1)] != '':
                s_list.append(s[i * j: i * (j + 1)])

        temp = s_list[0] # 반복되는지 확인할 문자
        temp_str = "" # 길이 측정을 위한 압축된 문자열
        count = 1# 문자가 반복하는 횟수

        for k in range(1, len(s_list)):
            if temp == s_list[k]: 
                count += 1
            else:
                if count > 1:
                    temp_str += str(count)
                temp_str += temp
                temp = s_list[k]
                count = 1

        if count > 1:
            temp_str += str(count)
        temp_str += temp

        answer = min(answer, len(temp_str))

    return answer

---

코드 설명

answer = len(s)

for i in range(1, len(s) // 2 + 1):

먼저 어떻게 문자열을 잘라도 압축되지 않는 경우가 있으므로 answer에 입력받은 문자열의 길이를 저장한다.

 

이후 문자열을 1개부터 문자열의 절반 길이(len(s) // 2)까지 의 단위까지 문자열을 압축해보면 된다.

 

s_list = []
for j in range(len(s) // i + 1):
    if s[i * j: i * (j + 1)] != '':
        s_list.append(s[i * j: i * (j + 1)])

문자열을 압축하기 전에 주어진 단위(i)로 문자열을 slicing 해준다.

먼저 slicing 된 문자열을 저장하기 위해 s_list를 선언해주고, 자른 문자열을 s_list에 차례대로 추가해준다.

 

temp = s_list[0] # 반복되는지 확인할 문자
temp_str = "" # 길이 측정을 위한 압축된 문자열
count = 1# 문자가 반복하는 횟수

for k in range(1, len(s_list)):
    if temp == s_list[k]:
        count += 1
    else:
        if count > 1:
            temp_str += str(count)
        temp_str += temp
        temp = s_list[k]
        count = 1

if count > 1:
    temp_str += str(count)
temp_str += temp

answer = min(answer, len(temp_str))

 

이제 문제에서 주어진 조건에 따라 문자열을 압축해 표현하면 된다.

 

비교할 단위 문자열을 temp에 저장하고 s_list에 있는 원소들과 비교해보면 된다.

 

같은 문자가 나오면 반복된 횟수(count)를 증가시켜주고  

다른 문자가 나오면 임시로 만든 압축 문자열(temp_str)에 count랑 temp를 붙여주면 된다.

count가 1인 경우(=반복된 문자가 없는 경우) 1은 생략한다고 했으므로 if문을 사용해 처리해준다.

 

주어진 문자열(s)의 마지막 문자도 붙여줘야 하므로 for문이 끝난 이후, 남은 문자(temp)도 count와 함께 붙여준다.

 

압축이 다 끝나면 기존에 있는 값과 비교해 작은 값을 answer에 저장해준다.