[Baekjoon] 1049번 기타줄

문제

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

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

입력

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

출력

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

예제

Example 1:

Input: 
4 2
12 3
15 4
Output: 
12

Example 2:

Input:
10 3
20 8
40 7
60 4
Output:
36

Example 3:

Input:
15 1
100 40
Output:
300

조건

시간 제한 : 2초

메모리 제한 : 128 MB

풀이과정

풀이

import sys
n, brand = map(int,sys.stdin.readline().split())
ans = []
brand_6 = []
brand_1 = []

for i in range(brand):
    p, q = map(int,sys.stdin.readline().split())
    brand_6.append(p)
    brand_1.append(q)
brand_6.sort()
brand_1.sort()

min_6 = brand_6[0]
min_1 = brand_1[0]

# 묶음만 산 경우
pakage_cost = ((n//6)+1) * min_6
ans.append(pakage_cost)
# 낱개만 산 경우
just_cost = n * min_1
ans.append(just_cost)
# 섞어서 산 경우
mix_cost = 0
while (n >= 6):
    n -= 6
    mix_cost += min_6
for i in range(n):
    mix_cost += min_1
ans.append(mix_cost)
print(min(ans))

간단한 문젠데 많이 틀렸다. 낱개만 산 경우를 고려하지 않았기 때문이다. 한편 , 각각 경우를 if문으로 나누고 섞어서 산 경우를 간단히 나타내면 다음과 같은 코드가 된다.

import sys
n, brand = map(int,sys.stdin.readline().split())
ans = []
brand_6 = []
brand_1 = []

for i in range(brand):
  p, q = map(int,sys.stdin.readline().split())
  brand_6.append(p)
  brand_1.append(q)

min_6 = min(brand_6)
min_1 = min(brand_1)

if min_6 < 6*min_1:
  if min_6 <(n%6)*min_1:
    # 패키지로 다사는 경우
    print((n//6)*min_6+min_6)
  else:
    # 패키지 + 낱개
    print((n//6)*min_6 + (n%6)*min_1)
else:
  # 낱개로 다사는 경우
  print(n*min_1)