코테 공부/백준

[Python/그리디] 2839 설탕 배달

prefer_all 2022. 8. 17. 12:46
문제

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

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

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

입력

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

출력

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

입력 예시 출력 예시
18 4
4 -1
6 2
9 3
11 3

풀이

내가 생각한 케이스
1. n이 5로 나누어 떨어짐 -> 바로 n//5가 정답
2. n이 5, 3 둘 다 사용해야 하는 케이스 
   -> 11의 경우 5+ 3*2 여서 5의 몫을 계산해버리면 5*2 + a 이렇게 되는 데 어떻게 처리할 지 문제 봉착
3. n이 5로는 나누어지지 않고 3으로만 나누어 지는 경우
   -> n//3이 정답

 

n = int(input())
ans = 0

# 3을 빼줘가면서 5의 배수가 되게 만든다
while n >= 0:
    if n%5 == 0:
        ans += n//5
        print(ans)
        break
    else: 
        n -= 3
        ans += 1
else: # (n이 4와 같을 때) else로 깔끔하게 처리 
    print(-1)

코드 참고

 

다른 풀이

if-else문으로 일단 5로 나누어지면 print, return

3으로 먼저 빼고 5로 나누어지는 지 체크

n = int(input())

if n % 5 == 0:  # 5으로 나눠떨어질 때
    print(n // 5)
else:
    p = 0
    while n > 0:
        n -= 3
        p += 1
        if n % 5 == 0:  # 3kg과 5kg를 조합해서 담을 수 있을 때
            p += n // 5
            print(p)
            break
        elif n == 1 or n == 2:  # 설탕 봉지만으로 나눌 수 없을 때
            print(-1)
            break
        elif n == 0:  # 3으로 나눠떨어질 때
            print(p)
            break

n이 5,3으로 표현될 수 없는 경우 -> n == 1 or n ==2 라고 표현