[프로그래머스] 이모티콘 할인행사 (Python)

link 🔗 https://school.programmers.co.kr/learn/courses/30/lessons/150368

프로그래머스

 

🔥 유저가 최대 100명, 이모티콘이 최대 7개로 범위가 그리 크지 않아 완전탐색으로도 시간초과가 나지 않았다.

🔥 하지만 삼중for문을 활용하지 않고도 풀 수 있지 않을까...🤔

 

from itertools import product

def solution(users, emoticons):
    answer = [0, 0]
    discount = [10, 20, 30, 40]
    emo_len = len(emoticons)
    # 중복 순열로 경우의 수 구하기
    pro_list = list(product(discount, repeat=len(emoticons)))
    
    for production in pro_list:
        # 이모티콘 플러스 가입자 수, 총 매출
        emoticon_plus, sales = 0, 0
        for user in users:
            limit_rate, limit_money = user[0], user[1]
            spent = 0
            for idx in range(emo_len):
                # 할인율이 유저 지정 할인율보다 크면 구매
                if production[idx] >= limit_rate:
                    spent += int(emoticons[idx] - emoticons[idx]*(production[idx]/100))
            # 총 사용한 금액이 유저 지정 총액 이상이면
            if spent >= limit_money:
                # 이모티콘 플러스 가입
                emoticon_plus += 1
            else:
                # 총액 미만이면 매출에 더함
                sales += spent
                
        # 이모티콘 플러스 가입자가 기존 최대값보다 더 많으면
        if emoticon_plus > answer[0]:
            # 이모티콘 플러스 가입자, 총 매출 모두 교체
            answer[0], answer[1] = emoticon_plus, sales
        # 이모티콘 플러스 가입자가 기존 최대값과 같고, 매출이 더 높으면
        elif emoticon_plus == answer[0] and sales > answer[1]:
            # 총 매출만 교체
            answer[1] = sales
                    
    return answer

 

product

- 파이썬 표준 라이브러리인 itertools에 있는 중복순열 구하는 함수

from itertools import permutations, combinations, product, combination_with_replacement

# 순열
permu_list = permutations(data, 5)

# 조합
comb_list = combinations(data, 5)

# 중복순열
product_list = product(data, repeat=5)

# 중복 조합
comb_rep_list = combinations_with_replacement(data, 5)