[프로그래머스] [3차] 압축

https://programmers.co.kr/learn/courses/30/lessons/17684

코딩테스트 연습 - [3차] 압축

 

우선, dic = {'A' : 1, 'B' : 2, ..., 'Z' : 26 } 를 만든다.

s = 0, e = 1 로 설정하고 msg = 'KAKAO' 를 예로 들면,

msg[s:e] 인 'K' 가 dic 에 있으면, e += 1 해서 'KA' 가 dic 에 존재하는지 비교한다.

dic 에 없으면 그 전의 비교 문자인 'K' 즉, dic[msg[s:e]] 를 answer 에 담고 dic 에 'KA' = 27(len(dic) + 1) 을 추가한다.

'K' 의 색인번호가 출력되었으니 다음 문자인 'A' (msg[e-1:e] 즉, s = e - 1) 부터 비교해나간다.

msg 의 모든 문자열 또는 문자를 비교하여 색인번호가 출력되었으면 break 하고 answer 을 반환한다. 

 

def solution(msg):
    answer = []
    dic = dict(zip('ABCDEFGHIJKLMNOPQRSTUVWXYZ', range(1, 27)))
    start, end = 0, 1
    
    while start < len(msg):
        if msg[start:end] in dic:
            if end >= len(msg): 
                answer.append(dic[msg[start:end]])
                break
            
            end += 1
        else:
            answer.append(dic[msg[start:end-1]])
            dic[msg[start:end]] = len(dic) + 1
            start = end - 1
            
    return answer

print(solution('KAKAO')) # [11, 1, 27, 15]