Programmers / 할인행사 / Python 파이썬

 

*문제 출처는 프로그래머스에 있습니다.

문제 제목: 할인행사 (2단계)

문제 사이트: https://school.programmers.co.kr/learn/courses/30/lessons/131127?language=python3 

프로그래머스

문제 설명
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
입출력 예
want number discount result
["banana", "apple", "rice", "pork", "pot"] [3, 2, 2, 2, 1] ["chicken", "apple", "apple", "banana", "rice", "apple", "pork", "banana", "pork", "rice", "pot", "banana", "apple", "banana"] 3
["apple"] [10] ["banana", "banana", "banana", "banana", "banana", "banana", "banana", "banana", "banana", "banana"] 0
입출력 예 설명
입출력 예 #1

문제 예시와 같습니다.
입출력 예 #2

사과가 할인하는 날이 없으므로 0을 return 합니다.

---

조건: 사길 원하는 물건이 판매하는 날로부너 10일 이내로 자신이 사고자 하는 것을 모두 살 수있어야함

할인하는 제품은 하루에 하나씩만 구매 가능

그 기간이 몇개나 되는지 카운트하고 리턴하면 되는 문제

 

나의 풀이

먼저 want의 아이템을 키로 하고, 갯수를 값으로 설정하여 딕셔너리를 만든다.

want_dict = {i:v for i,v in zip(want,number)}

입력

want, number

["banana", "apple", "rice", "pork", "pot"], [3, 2, 2, 2, 1]

출력결과

{'banana': 3, 'apple': 2, 'rice': 2, 'pork': 2, 'pot': 1}

 

그리고 판매하는 아이템들의 갯수를 세준다.

최대 10일까지만 보면 되므로 10개씩 끊어서 리스트를 생성 ( for문 사용)

그리고 해당 기간 내의 아이템들을 Counter() 함수를 통해 카운트 딕셔너리 형태로 반환한다.

tmp = discount[i:i+10]
tmp_dict = Counter(tmp)

그리고 사려고 했던 아이템이 해당 딕셔너리 안에 있는지 확인한다.

다음으로 해당 딕셔너리 안에서 사려고 했던 아이템의 개수가 동일한지 확인한다.

 

사실 아이템의 개수가 동일한지만 확인해보면 된다, (근데 난 왜 위에 것도 했지)

 all_items = False # 모든 아이템이 있는지 확인
all_counters = False # 해당 아이템의 수가 맞는지 확인
for i in want_dict:
    if i not in tmp:
        all_items = False
        break
    all_items = True

    if want_dict[i] != tmp_dict[i]:
        all_counters = False
        break
    all_counters = True

if all_items and all_counters:
    answer += 1

조건이 true가 되면 살 수 있는 방법에 하나를 추가한다.

전반적인 풀이 방법은 이렇고, 마지막으로 제일 위에 사려는 아이템이 판매하는 아이템의 리스트에 있는지 확인하고

want 리스트의 아이템이 discount 리스트에 없으면 0 을 반환한다.

탐색시간을 줄일 수 있다.

    if set(want) & set(discount) != set(want)  :
        return 0

두 리스트의 교집합이 want와 같지 않으면 0을 반환

 

from collections import Counter
def solution(want, number, discount):
    answer = 0
    l = len(discount)
    
    want_dict = {i:v for i,v in zip(want,number)}
    
    if set(want) & set(discount) != set(want)  :
        return 0
    
    else:
        for i in range(l):
            if i + 10 <= l:
                tmp = discount[i:i+10]
                tmp_dict = Counter(tmp)
                all_items = False # 모든 아이템이 있는지 확인
                all_counters = False # 해당 아이템의 수가 맞는지 확인
                for i in want_dict:
                    if i not in tmp:
                        all_items = False
                        break
                    all_items = True
                        
                    if want_dict[i] != tmp_dict[i]:
                        all_counters = False
                        break
                    all_counters = True
                    
                if all_items and all_counters:
                    answer += 1
                    

            else:
                if l - i >= len(want):
                    tmp = discount[i:]
                    tmp_dict = Counter(tmp)
                    all_items = False # 모든 아이템이 있는지 확인
                    all_counters = False # 해당 아이템의 수가 맞는지 확인
                    for i in want_dict:
                        if i not in tmp:
                            all_items = False
                            break
                        all_items = True

                        if want_dict[i] != tmp_dict[i]:
                            all_counters = False
                            break
                        all_counters = True

                    if all_items and all_counters:
                        answer += 1


    return answer

모범답안

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

 

코테를 오랜만에 풀어서 너무 길게 풀어졌나보다..하하

---