본문 바로가기

알고리즘 문제/백준

[파이썬, Python] 백준 1049 기타줄

기타줄 

문제

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

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

입력

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

출력

 

문제 풀이 방법

문제를 풀며 고려한 것은 다음과 같습니다.

1. 6개 혹은 1개의 가격 중에 0이 있는가

  > 해당 경우 몇개를 사도 가격이 0이 되기 때문에 바로 0을 리턴해줍니다.

2. 세트의 가격이 개별로 6개 사는 것 보다 비싼가

  > 해당 경우 전부 낱개로 사는 것이 싸기 때문에 낱개 곱하기 갯수로 리턴해줍니다.

3. 1~6개가 남았을 시 낱개로 사는 것이 싼가 아님 세트로 사는 것이 싼가

  > 6개에 10 / 1개에 3원인 경우 세트로 사고 남은 것이 4개인 경우 세트로 사는 것이 좋습니다.

 

코드는 아래와 같습니다.

package = []
single = []
n, m = map(int, input().split())
for i in range(m):
    a, b = map(int, input().split())
    if a == 0 or b == 0:
        print(0)
        exit()
    package.append(a)
    single.append(b)

package.sort() #최솟값 얻기
single.sort() #최솟값 얻기

if single[0] * 6 < package[0]:
    print(single[0] * n)
    exit()

sum1 = (n // 6) * package[0] + ((n % 6 + 6) // 6)* package[0]
sum2 = (n // 6) * package[0] + n % 6 * single[0] 


print(sum1 if sum1 < sum2 else sum2)