[프로그래머스 | Python] 할인 행사

프로그래머스_할인 행사

discount 리스트에서 원하는 것의 갯수만큼 확인하여, 원하는 list의 Counter와 동일하면 answer+1을 하고 아니면 그 다음 인덱스에서부터 원하는 것의 갯수를 확인한다

from collections import Counter
def solution(want, number, discount):
    answer = 0
    n = len(want)
    m = len(discount)
    buy = sum(number)
    
    want_list = []
    for i in range(n):
        want_list += [want[i]]*number[i]
    
    for i in range(m):
        if Counter(discount[i:i+buy])==Counter(want_list):
            answer += 1

    return answer