[프로그래머스/Lv.2] 할인 행사 | 파이썬 Python

문제

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

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

 

입출력 예

 

나의 풀이1 (실패)

def solution(want, number, discount):
    answer = 0
    
    for i in range(10):
        if discount[i] in want:
            index = want.index(discount[i])
            number[index] -= 1
    
    for i in range(10, len(discount)):
        if discount[i] in want:
            index = want.index(discount[i])
            number[index] -= 1
        
        if discount[i-10] in want:
            index = want.index(discount[i-10])
            number[index] += 1
       
        if all(x <= 0 for x in number):
            answer += 1
            
    return answer

• 할인하는 품목마다 `number`의 수를 `-1`해서, `number`의 모든 원소가 `0` 이하가 되면 제품을 모두 할인 받을 수 있는 경우라고 판단하는 방법으로 접근했다.
• for문을 10개만 먼저 실행하고, 10번째 이후부터는 현재 순서의 개수는 빼고, 10개 전의 개수는 더하는 방식으로 10개 묶음의 결과를 유지했다.
• 그런데 제출 결과 하나의 케이스에서만 결과가 틀렸다고 나왔다.

 

나의 풀이2 (통과)

def solution(want, number, discount):
    answer = 0
    
    for i in range(10):
        if discount[i] in want:
            index = want.index(discount[i])
            number[index] -= 1
    
    if all(x <= 0 for x in number):
            answer += 1
    
    for i in range(10, len(discount)):
        if discount[i] in want:
            index = want.index(discount[i])
            number[index] -= 1
        
        if discount[i-10] in want:
            index = want.index(discount[i-10])
            number[index] += 1
        
        if all(x <= 0 for x in number):
            answer += 1
            
    return answer

• 엣지케이스에 문제가 있을 거라는 생각에 다음과 같은 테스트 케이스를 추가해봤다.
  • `want = ["apple"]`, `number=[10]`, `discount=["apple", "apple", "apple", "apple", "apple", "apple", "apple", "apple", "apple", "apple", "apple", "apple"]`
  • 이 경우에 첫 번째 for문을 종료했을 때 조건을 달성하는데 답에 카운트되지 못한다.
• 그래서 두 개의 for문 사이에 첫 번째 for문의 결과를 카운트할 수 있는 if문을 추가해서 문제를 통과할 수 있었다.

 

다른 사람의 풀이

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

• 데이터를 딕셔너리 형태로 저장해서 비교하는 접근 방법
• 실행 속도 면에서는 내 풀이보다 두 배 정도 느리긴 한데, 가독성 면에서는 이 풀이가 훨씬 나은 거 같다.
• `Counter()`는 인자로 받은 리스트에서 각 원소가 몇 번씩 나오는지가 저장된 객체를 출력한다.
• 딕셔너리와 `Counter()`의 키와 값이 일치한다면 같은 것으로 판단하기 때문에 할인 조건을 충족했는지를 판단할 수 있다.

 

a = [1, 2, 3]
b = [1, 2, 3]

print(a == b)  # ✅ True (값이 같음)
print(a is b)  # ❌ False (다른 객체)

• 이 풀이를 보고 파이썬에서는 변수를 어떤 방식으로 비교하는지 궁금해서 찾아 봤다.
  • `==`: 값이 같은지를 비교한다.
  • `is`: 객체의 주소가 같은지를 비교한다.

 

회고

• 이번 기회에 참조값을 통해 값을 비교하는 자바스크립트와 값 자체를 통해서도 비교할 수 있는 파이썬의 차이점을 배울 수 있었다.
• `Counter()`도 오늘 제대로 사용해 봤는데 다음에도 문제 풀이에 적용해 보고 싶다!