[프로그래머스] LV.2 할인 행사(python)

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

프로그래머스

 

문제 설명

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

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

정현이가 원하는 제품을 나타내는 문자열 배열 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

 

문제 해결

from collections import Counter

def solution(want, number, discount):
    answer = 0
    dic = {i:0 for i in want}
    for i in range(len(want)):
        dic[want[i]] = number[i]
    i = 0
    while(i+10 <= len(discount)):
        flag = True
        dis = discount[i:i+10]
        arr = Counter(dis)
        for j in range(len(dic)):
            if dic[want[j]] > arr[want[j]]:
                flag = False
                break
        if(flag):
            answer += 1
        i += 1
    return answer

 

문제 풀이 설명

collections모듈의 Counter 클래스를 사용해 각 제품의 개수가 충족하는지 파악했다.

로직을 구성할 때 discount는 하루마다 무조건 10개씩 고정된 제품이 나오며, 해당 제품의 개수들이 내가 원하는 want 리스트에 각 number 리스트의 숫자와 대응하도록 하여 충족되는 날짜의 총 일수를 return하는게 중요하다고 생각했다.

---

1) want 리스트에 있는 제품을 기준으로 dictionary를 생성

2) 각 want 제품명을 키로, number 리스트의 개수를 값으로 대입

3) index를 0으로 선언 후, i + 10이 discount의 길이가 됐을 때 while문을 종료하도록 구성

4) while문을 돌면서 슬라이싱을 사용, discount 리스트를 10개씩 끊어 해당 날짜의 할인 제품목록을 구성

5) Counter 클래스를 사용해 개수를 dictionary로 반환, 내가 원하는 제품이 개수를 충족하면서 다 들어있는지 확인한다.

6) 만약 하나라도 모자란다면 flag 변수를 False로 바꾼 후 break를 사용해 for문을 탈출

7) 모든 제품이 할인한다면 flag 변수는 그대로 True로 나오게 되므로 answer 변수를 +1 해준다.

8) 모든 discount 리스트를 돌았다면 answer을 return하면 끝