[BOJ 1434번] 책 정리(Python)

📝 문제

💡 빈 박스 N개가 한 줄로 놓여져 있고, 박스에 책 M개를 넣으려고 한다.
박스는 1번부터 N번까지 번호가 매겨져 있고, 책도 1번부터 M번까지 번호가 매겨져 있다.

다음은 책을 박스에 넣는 방법이며, 가장 처음에는 1번 박스의 앞에 있고, 손에는 1번 책을 들고 있다.
현재 책이 현재 박스에 들어가지 않으면, 3번으로 간다. 아니면 2번으로 간다.
현재 책을 현재 박스에 넣는다.
다음 책을 손에 집고 1번으로 간다.
현재 박스를 다른 쪽으로 치운 다음에, 테이프로 못 열게 봉인한다.
다음 박스를 앞으로 가져오고 1번으로 간다.

i번 박스의 용량은 Ai이고, j번 책의 크기는 Bj이다.
책의 크기와 박스에 들어있는 책의 크기의 합이 박스의 용량을 넘지 않으면 책을 박스에 넣을 수 있다.
위의 방법대로 책을 넣었을 때, 전체 박스의 낭비된 용량의 합을 구하는 프로그램을 작성하시오.
박스의 낭비된 용량은 박스의 용량에서 박스에 들어있는 책의 크기의 합을 뺀 값이다.
입력으로 주어진 박스와 책의 순서를 변경하면 안된다.

⚙️ 입력 : 첫째 줄에 박스의 개수 N, 책의 개수 M이 주어진다.
둘째 줄에는 박스의 용량 A1, A2, ..., AN이 주어지고,
셋째 줄에는 B1, B2, ..., BM이 주어진다.

⚙️ 출력 : 첫째 줄에 전체 박스의 낭비된 용량의 합을 출력한다.

 

문제링크

1434번: 책 정리

---

✅ 풀이

n, m = map(int, input().split())
box = list(map(int, input().split()))
book = list(map(int, input().split()))

no = 0
for i in range(len(book)):
    if box[no] >= book[i]:
        box[no] -= book[i]
    else:
        while book[i] > box[no]:
            no += 1
        box[no] -= book[i]

print(sum(box))

💡 만약 박스의 크기가 책 크기보다 크다면, 박스에 책을 넣고 용량을 줄여주었다.
책 크기가 박스보다 더 큰 경우에는, 박스가 더 클때까지 이동하여 그 박스에 책을 넣어주도록 했다.