문제

문제 설명

XYZ 마트는 일정한 금액을 지불하면 10일 동안 회원 자격을 부여합니다. XYZ 마트에서는 회원을 대상으로 매일 한 가지 제품을 할인하는 행사를 합니다. 할인하는 제품은 하루에 하나씩만 구매할 수 있습니다. 알뜰한 정현이는 자신이 원하는 제품과 수량이 할인하는 날짜와 10일 연속으로 일치할 경우에 맞춰서 회원가입을 하려 합니다.

예를 들어, 정현이가 원하는 제품이 바나나 3개, 사과 2개, 쌀 2개, 돼지고기 2개, 냄비 1개이며, XYZ 마트에서 14일간 회원을 대상으로 할인하는 제품이 날짜 순서대로 치킨, 사과, 사과, 바나나, 쌀, 사과, 돼지고기, 바나나, 돼지고기, 쌀, 냄비, 바나나, 사과, 바나나인 경우에 대해 알아봅시다. 첫째 날부터 열흘 간에는 냄비가 할인하지 않기 때문에 첫째 날에는 회원가입을 하지 않습니다. 둘째 날부터 열흘 간에는 바나나를 원하는 만큼 할인구매할 수 없기 때문에 둘째 날에도 회원가입을 하지 않습니다. 셋째 날, 넷째 날, 다섯째 날부터 각각 열흘은 원하는 제품과 수량이 일치하기 때문에 셋 중 하루에 회원가입을 하려 합니다.

정현이가 원하는 제품을 나타내는 문자열 배열 want와 정현이가 원하는 제품의 수량을 나타내는 정수 배열 number, XYZ 마트에서 할인하는 제품을 나타내는 문자열 배열 discount가 주어졌을 때, 회원등록시 정현이가 원하는 제품을 모두 할인 받을 수 있는 회원등록 날짜의 총 일수를 return 하는 solution 함수를 완성하시오. 가능한 날이 없으면 0을 return 합니다.

제한사항

1 ≤ want의 길이 = number의 길이 ≤ 10
1 ≤ number의 원소 ≤ 10
number[i]는 want[i]의 수량을 의미하며, number의 원소의 합은 10입니다.
10 ≤ discount의 길이 ≤ 100,000
want와 discount의 원소들은 알파벳 소문자로 이루어진 문자열입니다.
1 ≤ want의 원소의 길이, discount의 원소의 길이 ≤ 12

풀이

아이디어

연속 10일동안 해당 조건을 만족하는지 확인한다.
이때 원하는 만큼의 항목과 가격을 dict로 넣어주고, 연속 10일의 항목을 해당 dict에서 개수를 뺀다.
그리고 나서 모든 값이 0 이면 answer += 1
그렇게 10개를 진행한 뒤, 10개 이후부터는 앞에 항목을 하나빼고, 뒤의 항목을 하나 추가하는 식으로 구현한다.
dict를 사용했으니 해당 항목을 찾을 때에는 O(1)이고 for문 1회 순회하기 때문에 O(discount)

예상 시간복잡도

O(discount - 10) == O(100000)

풀이시간

20분

실행시간

0.01ms~ 40.55ms

코드

def solution(want, number, discount):
    answer = 0
    dict = {}
    
    for w in range(len(want)) :
        dict[want[w]] = number[w] 
    
    for i in range(10) :
        if discount[i] in dict :
            dict[discount[i]] -= 1
            
    # dict가 다 0인지 확인
    if not any(dict.values()):
        answer += 1
        
    
    for j in range(10,len(discount)) :
        # 앞에꺼 빼기 
        if discount[j-10] in dict :
            dict[discount[j-10]] += 1
        # 뒤에꺼 넣기
        if discount[j] in dict :
            dict[discount[j]] -= 1
        # dict가 다 0인지 확인
        if not any(dict.values()):
            answer += 1
    
    return answer

다른 사람의 풀이

파이썬의 Counter 클래스를 활용해 더 간단히 풀 수 있다!!
이 Counter 클래스는 중복된 데이터의 개수를 dictionary 형태로 반환해준다.
또한 dictionary의 메서드들을 똑같이 사용할 수 있다.
따라서 아래 풀이는 Counter를 활용해 discount의 10개 연속된 값을 넣어주고, 그러면 알아서 dict형태로 개수를 세서 반환해준다.
이것과 정현이가 원하는 답을 dict에 넣어준 것과 형태가 같다면 answer += 1을 해준다!

from collections import Counter
def solution(want, number, discount):
    answer = 0
    dic = {}
    for i in range(len(want)):
        dic[want[i]] = number[i]

    for i in range(len(discount)-9):
        if dic == Counter(discount[i:i+10]): 
            answer += 1

    return answer