[Python] 프로그래머스, 할인 행사 Lv.2 (feat. zip, Counter, 파이썬)

반짝반짝 알고리즘, 인프준

 

 

 

 

 

 

 

 

 

 

할인 행사

문제 링크

프로그래머스

 

 

문제 설명

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 합니다.

 

 

 

 

 

 

---

 

 

 

 

 

 

해결 과정

zip() method와 Counter method를 활용해서 문제를 해결하였습니다.

 

1. 총 10일을 확인해야 하므로 discount의 길이에서 9를 뺀 값을 선언해 줍니다. ( 할인 목록 10개 - 9 = 1번 확인, 할인 목록 14 - 9 = 5번 확인 ), 그리고 반환할 날짜 days와 물건을 확인할 범위인 start와 end를 각각 0, 10으로 선언해 줍니다.

 

2. discount에서 9를 뺀 check 변수만큼 for loop을 돌려줍니다.

 

3. 반환할 days를 1 증가시켜 주고 Counter method로 discount 10일 치를 세어줍니다.

 

4. 제품이 담긴 want와 제품의 수량이 담긴 number를 zip method로 묶어주고 원소의 길이만큼 for loop을 돌려줍니다. 이때 제품과 수량이 counter에 모두 만족한다면 넘어가지만, 만족하지 않는 경우 days를 다시 1 감소시켜 줍니다.

 

5. 10일 범위의 확인을 완료한 후 start와 end를 1씩 증가시켜 마찬가지로 다음 날의 10일 범위를 확인해 줍니다.

 

6. 모든 확인이 끝나면 만족하는 날짜의 총 일수를 반환합니다.

 

 

 

제출 코드

from collections import Counter
def solution(want, number, discount):
    # 1
    check = len(discount) - 9
    days, start, end = 0, 0, 10
    
    # 2
    for _ in range(check):
        # 3
        days += 1
        counter = Counter(discount[start:end])
        
        # 4
        for item, count in zip(want, number):
            if counter[item] != count:
                days -= 1
                break
        
        # 5
        start += 1
        end += 1
    
    # 6
    return days

 

 

 

 

 

 

 

 

---

 

 

 

 

 

 

 

 

또 다른 풀이

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

 

 

비슷한 개념을 딕셔너리를 활용하여 해결한 코드입니다.

 

 

 

 

 

 

 

 

 

 

---

 

 

 

 

 

 

 

 

 

 

출처: 프로그래머스 코딩 테스트 연습,https://school.programmers.co.kr/learn/challenges