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

0. 문제 링크

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

프로그래머스

 

1. 문제 설명

1) 연속된 10일의 물품의 종류와 개수가

2) 주어진 물품의 종류(want)와 개수(number) 같은 날의 수 구하기

 

2. 입출력

# input
want = ["banana", "apple", "rice", "pork", "pot"]
number = [3, 2, 2, 2, 1]
discount = ["chicken", "apple", "apple", "banana", "rice", "apple", "pork", "banana", "pork", "rice", "pot", "banana", "apple", "banana"]

# output
result = 3

 

3. 코드

def solution(want, number, discount):
    answer = 0

    for i in range(len(discount)-9):
        sub = discount[i:i+10]                      # 10개씩 나누기
        flag = 0                                    # 개수가 동일한지 확인하는 변수
        for j in range(len(want)):                  # 물품 종류만큼 반복
            if sub.count(want[j]) != number[j]:     # 나눈 물품 종류의 개수와 원하는 물품 종류의 개수가 다르면
                flag = 1                            # flag를 1로 처리

        if flag == 0:                               # 개수가 모두 같으면
            answer += 1                             # 정답에 1 추가

    return answer

 

실행 결과

 

4. 회고

이 문제의 핵심은 'key-value' 이다.

처음에는 주어진 물품의 종류와 개수를 통해서 새롭게 배열을 생성했는데, count 함수를 이용하여 물품의 종류별 개수를 쉽게 셀 수 있는 것이 특징이다. 연속된 10일의 물품을 새로운 배열로 생성한 다음, 물품의 종류(want)를 기준으로 새로운 배열에서 물품의 개수를 구하여 물품의 개수(number)와 비교하는 방식으로 문제를 해결하였다.

 

5. Github

https://github.com/iihye/Algorithm/blob/main/Programmers/discount_offer.py

GitHub - iihye/Algorithm: Solving algorithm problems