프로그래머스 [Lv. 2] 할인 행사 {언어 : Python}

문제 링크

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

프로그래머스

정답 코드

from collections import defaultdict

def solution(want, number, discount):
    answer = 0
    tot = sum(number)
    lenW = len(want)
    lenD = len(discount)
    
    for i in range(lenD-tot+1):
        tenDict = defaultdict(int)
        isValid = True
        # 저장
        for j in range(i, min(i+10, lenD)):
            d = discount[j]
            tenDict[d] += 1
        # 확인
        for k in range(lenW):
            w = want[k]
            cntW = number[k]
            cntD = tenDict.get(w, 0)
            if cntW > cntD:
                isValid = False
                break
        
        if isValid:
            answer += 1
            
    return answer

풀이 방법

1. 10일 분량의 dictionary를 만들어서 discount 항목을 저장한다.
   1. defaultdict를 사용했기 때문에 해당 항목에 +1을 해주면서 수량을 파악할 수 있다.
2. 저장이 끝나면 want항목과 비교한다.
   1. 이 과정에서 유효한 경우만 answer +1을 한다.
3. answer를 반환한다.

느낀점

• 가장 단순한 접근이 때론 가장 효율적인 풀이일 수 있다.