Python

[백준-1434번] 책 정리

고고잉93 2023. 11. 17. 02:18
728x90

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

 

1434번: 책 정리

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

www.acmicpc.net











N, M = map(int, input().split())
box = list(map(int, input().split()))
book = list(map(int, input().split()))

idx = 0  # 박스 번호
for b in book:
    # 책이 박스에 들어갈 때까지 박스 교체
    while True:
        # 책이 박스에 들어감
        if b <= box[idx]:
            box[idx] -= b
            break

        # 책이 박스에 안 들어가면, 박스 교체
        idx += 1
print(sum(box))

 

머리는 이해했는데 ㅠㅠ 아직 생각대로 구현을 못하는것 같다..

 

 

728x90