[baekjoon] 백준 1817번(파이썬): 짐 챙기는 숌

문제

1817번: 짐 챙기는 숌

알고리즘

- 박스가 있는 경우와 없는 경우를 나눠 문제를 수행한다.

- 박스가 있는 경우 반복문을 통해 책의 무게를 확인한다.

- 박스에 넣을 수 있는 무게와 박스에 넣을 수 있는 최대 무게를 비교하여 박스에 넣을 수 있는 책의 개수를 카운트한다.

- 책의 무게를 다 확인 후에도 박스에 넣어야 하는 책이 있는 경우에 박스를 하나 더 카운트한다.

- 박스가 없는 경우는 0을 출력한다.

코드

import sys


n, m = map(int, sys.stdin.readline().split())

# 박스가 있다면
if n:
    k = list(map(int, sys.stdin.readline().split()))
    temp = 0 # 박스에 넣을 수 있는 무게
    cnt = 0 # 박스에 넣을 수 있는 책

    # 반복문을 통해 책의 무게를 확인
    for i in k:
        temp += i

        # 박스에 넣을 수 있는 무게와 박스에 넣을 수 있는 최대 무게가 같을 경우
        if temp == m:
            temp = 0
            cnt += 1

        # 박스에 넣을 수 있는 무게가 박스에 넣을 수 있는 최대 무게보다 무거울 경우
        elif temp > m:
            temp = i
            cnt += 1

    # 책의 무게를 다 확인 후에도 박스에 넣어야 하는 책이 있는 경우
    if temp:
        cnt += 1

    print(cnt)

# 박스가 없다면
else:
    print(0)

github

GitHub - junjange/CodingTest: 내가 푼 코딩 테스트 문제와 해결법