알고리즘

백준 #1049 - 기타줄

스히 2024. 3. 14. 10:37

 

문제

Day Of Mourning의 기타리스트 강토가 사용하는 기타에서 N개의 줄이 끊어졌다. 따라서 새로운 줄을 사거나 교체해야 한다. 강토는 되도록이면 돈을 적게 쓰려고 한다. 6줄 패키지를 살 수도 있고, 1개 또는 그 이상의 줄을 낱개로 살 수도 있다.

끊어진 기타줄의 개수 N과 기타줄 브랜드 M개가 주어지고, 각각의 브랜드에서 파는 기타줄 6개가 들어있는 패키지의 가격, 낱개로 살 때의 가격이 주어질 때, 적어도 N개를 사기 위해 필요한 돈의 수를 최소로 하는 프로그램을 작성하시오.

입력

첫째 줄에 N과 M이 주어진다. N은 100보다 작거나 같은 자연수이고, M은 50보다 작거나 같은 자연수이다. 둘째 줄부터 M개의 줄에는 각 브랜드의 패키지 가격과 낱개의 가격이 공백으로 구분하여 주어진다. 가격은 0보다 크거나 같고, 1,000보다 작거나 같은 정수이다.

출력

첫째 줄에 기타줄을 적어도 N개 사기 위해 필요한 돈의 최솟값을 출력한다.

예제 입력 1

4 2
12 3
15 4

예제 출력 1

12

예제 입력 2

 

10 3
20 8
40 7
60 4

예제 출력 2

36

예제 입력 3

15 1
100 40

예제 출력 3

300

예제 입력 4

17 1
12 3

예제 출력 4

36

예제 입력 5

7 2
10 3
12 2

예제 출력 5

12

예제 입력 6

9 16
21 25
77 23
23 88
95 43
96 19
59 36
80 13
51 24
15 8
25 61
21 22
3 9
68 68
67 100
83 98
96 57

예제 출력 6

6

 

 


 

 

먼저 문제의 조건을 살펴보자.

 

1) 필요한 기타줄의 수가 6개 이상이고, 6개 세트 최소 가격이 개별 최소 가격보다 작다면 세트를 선택한다.

2) 필요한 기타줄의 수가 6개 미만이지만, 6개 세트 최소 가격이 필요한 만큼만 개별 최소 가격보다 작다면 세트를 선택한다.

3) 이 외의 상황은 개별 최소 가격을 선택한다.

 

sets = []  # 세트 가격 저장할 리스트
indiv = []  # 개별 가격 저장할 리스트
money = 0  # 정답이자, 총 가격을 저장할 리스트

 

 

각 브랜드의 세트 가격과 개별 가격을 입력으로 받게 된다.

각 브랜드의 세트 가격은 sets 리스트에, 개별 가격은 indiv 리스트에 각각 저장하도록 했다.

이차원 배열이나 딕셔너리를 사용할 수도 있겠지만, 세트 가격의 최솟값, 개별 가격의 최솟값이 따로 필요했기 때문에 서로 다른 2개의 리스트를 사용하는 편이 효율적일 것이라 판단했다.

 

 

import sys
input = sys.stdin.readline

n, m = map(int, input().split())
sets = []  # 세트 가격 저장할 리스트
indiv = []  # 개별 가격 저장할 리스트
money = 0  # 정답이자, 총 가격을 저장할 리스트 

for _ in range(m):  # 기타줄 정보 입력받아 각 리스트에 분리해서 저장하기
    a, b = list(map(int, input().split()))
    sets.append(a)
    indiv.append(b)

while n != 0:  # 남은 돈이 0이라면 멈추기
    if n >= 6 and min(sets) <= min(indiv) * 6:  # 필요한 기타줄이 6 이상이고, 세트 가격이 더 싸다면
        money += min(sets)  # 세트 가격으로 구매
        n -= 6  # 구매한 만큼 필요한 기타줄 차감
    elif n < 6 and min(indiv) * n >= min(sets):  # 필요한 기타줄이 6 미만이지만 그래도 세트 가격이 더 싸다면
        money += min(sets)  # 세트 가격으로 구매
        n = 0  # 모두 구매한 상황이므로 필요한 기타줄 0
    else:  # 이 외의 경우라면
        money += min(indiv) * n  # 개별 가격으로 구매
        n = 0  # 모두 구매한 상황이므로 필요한 기타줄 0

print(money)

 

 

이렇게 문제의 조건을 while문 안에 삽입시켜 코드를 완성했다.