[ 프로그래머스 / LV2 / 파이썬 Python ] 이모티콘 할인행사

링크 : https://school.programmers.co.kr/learn/courses/30/lessons/150368

프로그래머스

 

코드

---

from itertools import product

def getPrice(emoticons,ratio,userRate):
    return sum([(emoticons[i]*(100-ratio[i])//100) for i in range(len(emoticons)) if ratio[i] >= userRate])

def solution(users, emoticons):
    answer = []
    for ratio in product([10,20,30,40],repeat = len(emoticons)):
        membership,total = 0,0
        for user in users:
            if (res:=getPrice(emoticons,ratio,user[0])) >= user[1]:
                membership += 1
                continue
            total += res
        answer.append([membership,total])
    return sorted(answer,reverse = True)[0]

 

풀이

---

문제를 보고 고민해보았다면 아마도 완전탐색을 고려해보았을 것이다.

이모티콘의 개수는 7개, 할인률은 4개이기 때문에 모든 경우를 고려해도 4**7 == 만어쩌구 밖에 되지 않기 때문에 모든 조합을 만들어서 모든 경우를 따져보는 완전탐색을 사용해야 한다.

 

나는 itertools의 product라는 메서드를 사용했다.

product 메서드는 repeat이라는 옵션을 주어 repeat에 주어지는 값만큼 원소의 개수를 중복조합으로 생성한다.

 

따라서 할인율을 이모티콘의 개수만큼 중복조합으로 생성하고 유저마다 해당 할인율 조합으로 멤버십가입 또는 구매를 결정해준다.

모든 경우를 answer에 저장하고 역정렬 후 가장 첫 원소를 리턴해주면 된다.