백준 문제 중 2839번, 그리디, ‘설탕배달’

백준 문제 중 2839

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

문제

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

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

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

입력

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

출력

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

---

풀이

kgs = int(input())

# return을 활용 하기위해 함수로 구현
def min_bags(kgs):
  
    #5kg봉지가 최대한 많을 수록 i+j가 적으니
    #kgs//5부터 0까지 차례로 loop
    for i in range(kgs//5,-1,-1):
        
        # kgs가 5의 배수라면 바로 i출력 
        if 5*i==kgs:
            return print(i)
        
         #3kg봉지는 적을수록 i+j가 
         #적으니 1에서 1//kgs순으로 loop
        for j in range(1,kgs//3+1):
            if (5*i+3*j)==kgs:
                return print(i+j)
   
 '''return으로 빠져나오지 못할 
    경우 불가능한 경우 이므로 -1출력'''
    return print(-1)

min_bags(kgs)

12
4

배운점

1. 처음 코드를 짤때 최솟값을 저장하는 min변수와 tmp변수를 선언해 tmp에 i+j를 저장해 min과 비교해가며 코드를 짜려고 했는데 return을 활용하는게 더 효율적 이라는걸 알았다.

2. 다른 사람 코드를 보니 더 효율적으로 클린한 코드를 짠거 같아 아래 코드도 학습했다.

• 다른 사람이 짠코드와 비교

num = int(input()) 
count = 0 
while num >= 0: 
    if num % 5 == 0: 
        count += int(num // 5) 
        print(count) 
        break

    num -= 3
    count += 1 
else: print(-1) 풀이를