[2020카카오인턴십] 보석 쇼핑 (python 파이썬)

풀이 보면 쉬운데

왜 못풀까 ㅠㅠㅠㅠㅠ 이 문제 덕에 딕셔너리 사용법을 알았다 그거로 만족하자 .. ^^

 

문제

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

코딩테스트 연습 - 보석 쇼핑

 

풀이

딕셔너리를 사용한다 !

주어진 배열이 ['AA', 'AB', 'AC', 'AA', 'AC'] 라고 하자.

L, R 값을 각각 AA를 가르키도록 초기화를 시켜준다. (L = R = 0)

dic = {gems[0]:1} 함으로써 딕셔너리 초기화 (AA:1)

1. L~R 범위 내 배열의 요소가 전체 보석 종류와 일치하면 더 좋은 답인지 체크 후, R += 1

2. L~R 범위 내 배열의 요소가 전체 보석 종류와 일치하지 않으면 L += 1 한다음 벗어난 범위에 대한 딕셔너리 카운트 값 정리 ( 카운트가 1개였으면 딕셔너리 삭제, 1개 초과였으면 -1 해주면 된다.)

3. L 또는 R 값이 배열의 끝을 가르키면 break

소스코드

def solution(gems):
    answer = [0, len(gems)-1]
    l, r = 0, 0
    en = len(set(gems))
    n = len(gems)
    dic = {gems[0]:1}
    
    while True:
        if l == n or r == n:
            break
        if len(dic)==en:
            if r-l<answer[1]-answer[0]:
                answer = [l, r]
            if dic[gems[l]]==1:
                del dic[gems[l]]
            else:
                dic[gems[l]] -= 1
            l += 1
        else:
            r += 1
            if r == n:
                break
            if gems[r] in dic.keys():
                dic[gems[r]] += 1
            else:
                dic[gems[r]] = 1
    return [answer[0]+1, answer[1]+1]