[백준 알고리즘][파이썬/Python] 2839번: 설탕 배달

문제

상근이는 요즘 설탕공장에서 설탕을 배달하고 있다. 상근이는 지금 사탕가게에 설탕을 정확하게 N킬로그램을 배달해야 한다. 설탕공장에서 만드는 설탕은 봉지에 담겨져 있다. 봉지는 3킬로그램 봉지와 5킬로그램 봉지가 있다.

 

상근이는 귀찮기 때문에, 최대한 적은 봉지를 들고 가려고 한다. 예를 들어, 18킬로그램 설탕을 배달해야 할 때, 3킬로그램 봉지 6개를 가져가도 되지만, 5킬로그램 3개와 3킬로그램 1개를 배달하면, 더 적은 개수의 봉지를 배달할 수 있다.

 

상근이가 설탕을 정확하게 N킬로그램 배달해야 할 때, 봉지 몇 개를 가져가면 되는지 그 수를 구하는 프로그램을 작성하시오.

 

입력

첫째 줄에 N이 주어진다. (3 ≤ N ≤ 5000)

 

출력

상근이가 배달하는 봉지의 최소 개수를 출력한다. 만약, 정확하게 N킬로그램을 만들 수 없다면 -1을 출력한다.

 

입출력 예

 

제출 코드

n = int(input())

if n % 5 == 0:
    print(n//5)
    
else:
    lst = []
    i = 1

    while n >= 3:
        n -= 3
        if n % 5 == 0:
            lst.append(i + n//5)
        i += 1

    if len(lst) == 0:
        print(-1)
    else:
        print(min(lst))

(1) line 3, 4

운반해야 하는 설탕의 무게가 5의 배수라면 5kg 봉지에 담아 배달하는 게 최소 개수이다. 그러므로 입력값이 5의 배수라면 어떠한 조치도 취할 필요 없이 바로 print(n//5)를 해버리자.

 

(2) line 7, 8

리스트를 하나 선언하는데 아래 로직에서 더 자세하게 설명하겠지만 "3kg와 5kg봉지로 나누어 배달 가능한 경우의 설탕 봉지 숫자"를 담을 것이다. i는  3kg 설탕의 봉지 개수를 의미한다.

 

(3) line 10 ~ 14

line 11에서 전체 설탕 무게에서 3kg을 뺀다. 즉, 설탕을 3kg 봉지에 하나 담았다고 생각하자. 이때 n이 5의 배수라면 배달 가능한 경우이므로 리스트에 3kg 봉지의 개수와 5kg 봉지의 개수를 합산한 값을 append한다.

 

이후 3kg 봉지를 하나씩 늘려보며 배달 가능한 경우의 설탕 봉지 개수를 계속 리스트에 append한다. 여기서 주의해야 할 점은 while의 조건문이 n >= 3이라는 것이다. n이 3 미만이라면 3kg 봉지로도 담을 수 없기 때문에 무시하면 된다.

 

(4) line 16 ~ 19

만약 리스트가 비어있다면 배달 가능한 경우가 한 번도 없었다는 소리다. '출력'에 대한 설명에서 "정확하게 N킬로그램을 만들 수 없다면 -1을 출력"하라고 했으므로 이 경우에는 -1을 출력한다.

 

반면 리스트에 하나 이상의 원소가 담겼을 경우, 리스트 내 원소 중 가장 작은 값을 출력한다.

 

 

 

Source : https://www.acmicpc.net/problem/2839