프로그래머스 레벨3 - 카카오 2020인턴 보석쇼핑

문제주소

programmers.co.kr/learn/courses/30/lessons/67258

내 풀이

def solution(gems):
    items = len(set(gems))
    shopping = dict()
    start = 0
    end = 0
    candidates = []
    for i in range(len(gems)):
        shopping[gems[i]] = shopping.get(gems[i], 0) + 1
        end = i
        if len(shopping) == items:
            for j in range(start, len(gems)):
                # start를 현재 j 값으로 셋팅하고
                start = j
                candidates.append((end - start, [start + 1, end + 1]))
                # 다음을 위해 다음 값으로 변경해줌
                start = j+1
                shopping[gems[j]] = shopping.get(gems[j]) - 1
                if shopping[gems[j]] == 0:
                    del (shopping[gems[j]])
                    break
    candidates.sort(key=lambda x: (x[0], x[1][0]))
    return candidates[0][1]

예전에 2일에 걸쳐서 고민했던 문제라.. 그 풀이가 생각나서 접근방식은 쉽게 작성했다. 다만 분명이 맞는 접근인데 오답 처리되는게 있어서 start를 셋팅해주는 과정을 break 되는 것을 고려해서 j+1로 다음 값을 설정해주니 통과되었다.

다른 사람 풀이

for문 안에 작은 for문으로 start값을 올려주었는데 while으로 if-else문으로 작성해준 풀이도 있었다. 더 짧게 if-else 문으로 푼 풀이들이 있어서 인상깊었다. 이런 풀이들은 answer에서 조건 비교 후에 바로 넣어주었다. 나는 아직 모든 조건을 구하고 해당 조건을 sort해서 작성하는 것이 케이스를 빠뜨리는 경우를 차단할 수 있어서 아직은 candidates를 구하는 것을 선호한다.

정리

옛날에 풀었던 풀이랑 방식은 같지만 그때보단 좀 더 깔끔하게 작성한 것 같아서 발전했다고 생각한다. 😀