[BOJ] 그리디 알고리즘 - 설탕배달

 

문제

 

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

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

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

 

입력

 

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

 

출력

 

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

 

 

해설 코드

sugar = int(input())
bag = 0

while sugar >= 0:
  # sugar가 5로 나뉘어지면 그대로 bag를 구함 
  if sugar % 5 ==0:
    bag += (sugar//5)
    print(bag)
    break
  # sugar가 5로 나뉘어지지 않는 경우, (sugar - 3) 을 반복하여 5의 배수가 될 때까지 반복
  sugar -= 3
  # 3을 빼면 그만큼 3kg 봉지 개수가 늘어나므로 +1 을 함
  bag += 1 

# sugar 값이 0보다 작아지는 경우는 -1 을 출력
else:
  print(-1)

이 코드의 전체적인 전개 방식은 입력된 sugar 값이 5의 배수가 될 때까지 3을 빼주는 행위를 반복하는 것이다. 

 

 

오답 이유 및 유의점

 

이 문제를 처음 풀이했을 때, 단순하게 각 조건에 따른 조건문을 전부 적어서 문제를 해결하는 방식으로 풀려고 했었다. 그래서 중간 중간 예시 입출력 값과는 다른 값이 나왔다. 그러기에 단순히 조건을 나열하는 방식을 바로 생각하기보다는, 어떻게 하면 값을 구하는 과정을 일반화된 공식으로 만들 수 있을지에 대해서 고민을 한 후에 코드를 작성해야겠다는 생각이 들었다.