[프로그래머스 Lv.2] 2023 카카오 블라인드 채용 - 이모티콘 할인행사

문제

카카오톡에서는 이모티콘을 무제한으로 사용할 수 있는 이모티콘 플러스 서비스 가입자 수를 늘리려고 합니다.

 

이를 위해 카카오톡에서는 이모티콘 할인 행사를 하는데, 목표는 다음과 같습니다.

1. 이모티콘 플러스 서비스 가입자를 최대한 늘리는 것.
2. 이모티콘 판매액을 최대한 늘리는 것.

1번 목표가 우선이며, 2번 목표가 그 다음입니다.

 

이모티콘 할인 행사는 다음과 같은 방식으로 진행됩니다.

• n명의 카카오톡 사용자들에게 이모티콘 m개를 할인하여 판매합니다.
• 이모티콘마다 할인율은 다를 수 있으며, 할인율은 10%, 20%, 30%, 40% 중 하나로 설정됩니다.

 

카카오톡 사용자들은 다음과 같은 기준을 따라 이모티콘을 사거나, 이모티콘 플러스 서비스에 가입합니다.

• 각 사용자들은 자신의 기준에 따라 일정 비율 이상 할인하는 이모티콘을 모두 구매합니다.
• 각 사용자들은 자신의 기준에 따라 이모티콘 구매 비용의 합이 일정 가격 이상이 된다면, 이모티콘 구매를 모두 취소하고 이모티콘 플러스 서비스에 가입합니다.

 

카카오톡 사용자 n명의 구매 기준을 담은 2차원 정수 배열 users, 이모티콘 m개의 정가를 담은 1차원 정수 배열 emoticons가 주어집니다.

이때, 행사 목적을 최대한으로 달성했을 때의 이모티콘 플러스 서비스 가입 수와 이모티콘 매출액을 1차원 정수 배열에 담아 return 하도록 solution 함수를 완성해주세요.

 

코드

from itertools import product

def solution(users, emoticons):
    answer = []
    
    for pro in product([10, 20, 30, 40], repeat=len(emoticons)):
        new_emoticons = [int((100-a) * b * 0.01) for a, b in zip(pro, emoticons)]
        people = 0
        total = 0
        
        for user in users:
            wish_list = []
            for i, p in enumerate(pro):
                if p >= user[0]: # user가 원하는 할인율을 넘었을 때
                    wish_list.append(new_emoticons[i])
            
            if sum(wish_list) >= user[1]: # user의 최대 한도를 넘겼을 경우
                people += 1
            else:
                total += sum(wish_list)
                
        answer.append([people, total])
    
    return sorted(answer, key = lambda x : (-x[0], -x[1]))[0]