코딩 테스트 연습) 2023 KAKAO BLIND RECRUITMENT 이모티콘 할인행사(python3)

[문제 설명]

카카오톡에서는 이모티콘을 무제한으로 사용할 수 있는 이모티콘 플러스 서비스 가입자 수를 늘리려고 합니다.
이를 위해 카카오톡에서는 이모티콘 할인 행사를 하는데, 목표는 다음과 같습니다.

이모티콘 플러스 서비스 가입자를 최대한 늘리는 것.
이모티콘 판매액을 최대한 늘리는 것.
1번 목표가 우선이며, 2번 목표가 그 다음입니다.

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

n명의 카카오톡 사용자들에게 이모티콘 m개를 할인하여 판매합니다.
이모티콘마다 할인율은 다를 수 있으며, 할인율은 10%, 20%, 30%, 40% 중 하나로 설정됩니다.
카카오톡 사용자들은 다음과 같은 기준을 따라 이모티콘을 사거나, 이모티콘 플러스 서비스에 가입합니다.

각 사용자들은 자신의 기준에 따라 일정 비율 이상 할인하는 이모티콘을 모두 구매합니다.
각 사용자들은 자신의 기준에 따라 이모티콘 구매 비용의 합이 일정 가격 이상이 된다면, 이모티콘 구매를 모두 취소하고 이모티콘 플러스 서비스에 가입합니다.
다음은 2명의 카카오톡 사용자와 2개의 이모티콘이 있을때의 예시입니다.

 

사용자	비율	가격
1	40	10,000
2	25	10,000

 

이모티콘	가격
1	7,000
2	9,000

 

1번 사용자는 40%이상 할인하는 이모티콘을 모두 구매하고, 이모티콘 구매 비용이 10,000원 이상이 되면 이모티콘 구매를 모두 취소하고 이모티콘 플러스 서비스에 가입합니다.
2번 사용자는 25%이상 할인하는 이모티콘을 모두 구매하고, 이모티콘 구매 비용이 10,000원 이상이 되면 이모티콘 구매를 모두 취소하고 이모티콘 플러스 서비스에 가입합니다.

1번 이모티콘의 가격은 7,000원, 2번 이모티콘의 가격은 9,000원입니다.

만약, 2개의 이모티콘을 모두 40%씩 할인한다면, 1번 사용자와 2번 사용자 모두 1,2번 이모티콘을 구매하게 되고, 결과는 다음과 같습니다.

사용자	구매한 이모티콘	이모티콘 구매 비용	이모티콘 플러스 서비스 가입 여부
1	1, 2	9,600	X
2	1, 2	9,600	X

 

이모티콘 플러스 서비스 가입자는 0명이 늘어나고 이모티콘 판매액은 19,200원이 늘어납니다.
하지만, 1번 이모티콘을 30% 할인하고 2번 이모티콘을 40% 할인한다면 결과는 다음과 같습니다.

사용자	구매한 이모티콘	이모티콘 구매 비용	이모티콘 플러스 서비스 가입 여부
1	2	5,400	X
2	1, 2	10,300	O

2번 사용자는 이모티콘 구매 비용을 10,000원 이상 사용하여 이모티콘 구매를 모두 취소하고 이모티콘 플러스 서비스에 가입하게 됩니다.
따라서, 이모티콘 플러스 서비스 가입자는 1명이 늘어나고 이모티콘 판매액은 5,400원이 늘어나게 됩니다.

카카오톡 사용자 n명의 구매 기준을 담은 2차원 정수 배열 users, 이모티콘 m개의 정가를 담은 1차원 정수 배열 emoticons가 주어집니다. 이때, 행사 목적을 최대한으로 달성했을 때의 이모티콘 플러스 서비스 가입 수와 이모티콘 매출액을 1차원 정수 배열에 담아 return 하도록 solution 함수를 완성해주세요.

---

[제한사항]

• 1 ≤ users의 길이 = n ≤ 100
  • users의 원소는 [비율, 가격]의 형태입니다.
  • users[i]는 i+1번 고객의 구매 기준을 의미합니다.
  • 비율% 이상의 할인이 있는 이모티콘을 모두 구매한다는 의미입니다.
    • 1 ≤ 비율 ≤ 40
  • 가격이상의 돈을 이모티콘 구매에 사용한다면, 이모티콘 구매를 모두 취소하고 이모티콘 플러스 서비스에 가입한다는 의미입니다.
    • 100 ≤ 가격 ≤ 1,000,000
    • 가격은 100의 배수입니다.
• 1 ≤ emoticons의 길이 = m ≤ 7
  • emoticons[i]는 i+1번 이모티콘의 정가를 의미합니다.
  • 100 ≤ emoticons의 원소 ≤ 1,000,000
  • emoticons의 원소는 100의 배수입니다.

---

[입출력 예]

 

users	emoticons	result
[[40, 10000], [25, 10000]]	[7000, 9000]	[1, 5400]
[[40, 2900], [23, 10000], [11, 5200], [5, 5900], [40, 3100], [27, 9200], [32, 6900]]	[1300, 1500, 1600, 4900]	[4, 13860]

---

[코드]

def solution(users, emoticons):
    answer = [0, 0]
    data = [10, 20, 30, 40]
    discount = []
    def dfs(tmp, d):
        if d == len(tmp):
            discount.append(tmp[:])
            return
        else:
            for i in data:
                tmp[d] += i
                dfs(tmp, d+1)
                tmp[d] -= i
    dfs([0]*len(emoticons), 0)

    for disc in discount: 
        cnt = 0
        get = 0
        for i in users:
            pay = 0
            for j in range(len(disc)):
                if i[0] <= disc[j]:
                    pay += emoticons[j] * (100 - disc[j])/100
                if pay >= i[1]:
                    break
            if pay >= i[1]: 
                pay = 0
                cnt += 1
            get += pay
        if cnt >= answer[0]:
            if cnt == answer[0]:
                answer[1] = max(answer[1], get)
            else:
                answer[1] = get
            answer[0] = cnt

    return answer

 

 

---

[코드 설명]

1. solution 함수는 users와 emoticons 두 개의 매개변수를 받습니다. users는 각 사용자의 정보를 포함하는 리스트이고, emoticons는 각 이모티콘의 가격을 나타내는 리스트입니다.
2. answer라는 변수는 결과를 저장하는데 사용됩니다. 초기값으로 [0, 0]이 설정되어 있습니다. answer[0]은 최대 구매 가능한 이모티콘 수, answer[1]은 해당 이모티콘을 판매하여 얻을 수 있는 이모티콘의 양을 나타냅니다.
3. data 리스트에는 가능한 할인율이 들어있습니다. ([10, 20, 30, 40])
4. discount라는 리스트는 모든 가능한 할인율 조합을 저장하는데 사용됩니다. dfs 함수는 재귀적으로 할인율을 조합하고, 이를 discount 리스트에 추가합니다.
5. 이후 for disc in discount: 루프에서는 각 할인율 조합에 대해 다음을 수행합니다.
   • cnt 변수는 현재 할인율 조합으로 얻을 수 있는 최대 구매 가능한 이모티콘 수를 나타냅니다.
   • get 변수는 해당 이모티콘을 판매하여 얻을 수 있는 이모티콘의 양을 누적합니다.
6. for i in users: 루프에서는 각 사용자에 대해 다음을 수행합니다.
   • pay 변수는 현재 사용자에게 지불해야 할 금액을 나타냅니다.
   • for j in range(len(disc)): 루프에서는 각 할인율에 대해 다음을 수행합니다.
     • 사용자의 제한 금액보다 작거나 같은 할인율을 찾아서, 해당 할인율로 이모티콘을 구매할 경우의 지불 금액을 계산합니다.
     • pay에 계산된 금액을 누적합니다.
     • 만약 pay가 사용자의 제한 금액을 초과하면 루프를 종료합니다.
7. 만약 pay가 사용자의 제한 금액을 초과하면, pay를 0으로 초기화하고 cnt를 증가시킵니다. 이는 사용자가 제한 금액을 초과할 경우 추가로 이모티콘을 무료로 받을 수 있는 횟수를 뜻합니다.
8. get은 현재 할인율 조합으로 얻을 수 있는 이모티콘의 양을 누적하게 됩니다.
9. 최종적으로 현재 할인율 조합이 이전에 계산된 최대 구매 가능 이모티콘 수보다 크거나 같으면서, 그 값이 현재 최대값보다 크다면 answer를 갱신합니다.
10. 마지막으로 answer를 반환합니다.