[4코1파] #58. LV 2. 할인 행사

[4코1파] 4명의 안드로이드 개발자와 1명의 파이썬 개발자의 코딩 테스트 서막 : 4코1파

Rule :

하루에 1문제씩 풀기.
한 문제당 30분씩은 고민하기.
왜 그렇게 풀었는지 공유하기.
하루라도 놓친다면 벌금은 1,000원

START :

[3코1파] 2023.01.04~ (58일차)
[4코1파] 2023.01.13~ (49일차)

Today :

2023.03.02 [58일차]

프로그래머스 LV 2
할인 행사
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

입출력 예

입출력 예 설명

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

입출력 예 #2
사과가 할인하는 날이 없으므로 0을 return 합니다.

풀이방법

요것은 리스트와 리스트에 해당하는 number가 담긴 리스트가 나온 문제로, 주어진 다른 리스트와 비교하는건 dictionary랑 Counter로 거의 끝나는 문제..
렙 투에서 가장 쉬운 문제가 아닐까 싶음

일단 want와 상응하는 number를 매칭시켜서 제품에 해당하는 개수를 dictionary 로 바꿔주었고,

가게에서 할인하는 discount 리스트를 돌면서,
정현이는 시작하는 날부터 순서대로 10일까지만 확인하면되므로
idx 기준으로 idx+10 까지 인덱싱을 한다.
이를 Counter 함수로 각 제품 key와 그의 값 value가
위에서 만든 want dictionary랑 비교해서 같으면 answer +=1
아니면 pass

쓸데없는 계산과정을 줄이기 위해서
discount에 원소가 want dictionary에 없으면 continue,
길이가 10미만이면 break를 거는 로직을 추가 !

증빙

내 코드

from collections import Counter
def solution(want, number, discount):
    answer = 0
    want_dic = {w:n for w,n in zip(want,number)}
    
    for idx in range(len(discount)):
        
        if discount[idx] not in want_dic.keys():
            continue
        if len(discount[idx:idx+10]) < 10:
            break
        
        if want_dic == dict(Counter(discount[idx:idx+10])):
            answer+=1
    
    return answer

여담

코드 실행속도가 좀 걸리긴하지만
누가뭐래도 내새끼