[Python] 점프 투 파이썬_종합문제_Q14_문자열 압축하기

문제

https://wikidocs.net/17114

08장 종합문제

 

점프 투 파이썬 Q14 문제의 지문은 다음과 같다.

Q14 문자열 압축하기
문자열을 입력받아 같은 문자가 연속적으로 반복되는 경우에 그 반복 횟수를 표시해 문자열을 압축하여 표시하시오.

입력 예시: aaabbcccccca
출력 예시: a3b2c6a1

---

작성 코드

아래는 Q14 문제를 해결하기 위해 작성한 코드이다.

s = 'aaabbcccccca' # 입력값

rst = ''
cnt = 1
for i in range(1, len(s)-1): # 'string index out of range' 에러를 방지하기 위해 -1함
    if s[i] != s[i+1]:
        rst += s[i]
        rst += str(cnt) # rst는 문자열 변수 이기 때문에 str을 이용해 cnt의 자료형을 문자열로 변경함
        cnt = 1
    else:
        cnt += 1

# 'string index out of range' 에러를 방지하기 위해 -1했던 for문에서 누락되었던 문자(character) 및  cnt를 rst에 추가함
rst += s[-1]
rst += str(cnt)

print(rst)

---

문제와 해결

문제를 잘 풀었다고 생각했는데,

다음과 같이 6과 1 사이에 a가 누락되어 출력되었다.

a3b2c61

 

for문 안에 조건을 하나 더 넣어야 하나 고민하고 이런 저런 시도를 했었다.

그러다 보니 코드가 점점 심각해졌다.

그런데,

코드 마지막 부분에 짧은 한 줄 넣어주면 간단히 해결 될 문제였다.

rst += s[-1]

 

허허...