1049 기타줄

https://www.acmicpc.net/problem/1049

1049번: 기타줄

 

내 풀이(맞음)

n, m = map(int, input().split())
p6 = []
p1 = []
for i in range(m):
    a, b = map(int, input().split())
    p6.append(a)
    p1.append(b)
p6.sort()
p1.sort()

if(n>=6):
    price1 = p6[0] * (n//6)
    price2 = price1
    price1 += p6[0]
    price3 = p1[0] * n
    
    n %= 6
    if(n > 0):
        price2 += p1[0] * n
    print(min(price1, price2, price3))
else:
    s1 = p1[0] * n
    s6 = p6[0]
    price = min(s1, s6)
    print(price)

p6 리스트엔 6개 패키지 가격을 넣고 p1엔 낱개 가격을 넣은 다음 오름차순으로 정렬한다. 그리고 끊어진 줄의 개수인 n이 6보다 크면 price1, price2, price3을 만든다. price1에는 끊어진 줄의 개수 보다 많이 사더라도 6개 패키지 가격으로 산 가격을 넣는다. price2에는 6개 패키지와 낱개를 같이 산 가격을 넣는다. price3에는 낱개로만 산 가격을 넣는다. 그리고 min 함수를 통해 최소 값을 출력한다.

만약 끊어진 줄의 개수가 6개보다 작으면 s1에 낱개로 산 가격, s2에 패키지로 산 가격을 넣고 비교해서 최소값을 출력한다.