백준 BAEKJOON 1434번 책 정리 [PYTHON/파이썬]

 

 

 

 

 

 

백준 BAEKJOON 1434번 책 정리 [PYTHON/파이썬]

---

<문제 출처> (BRONZE Ⅱ)

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

 

 

 

 

 

 

 

 

<풀이>

박스의 인덱스를 i, 책의 인덱스를 j로 설정하고

while 문을 책의 개수만큼 돌렸다.

현재 들고 있는 책이 앞에 있는 박스의 용량보다 같거나 작다면

책을 해당 박스에 집어 놓고 박스의 용량에서 책의 크기만큼 빼줬다.

그리고 만약 책이 박스의 용량보다 크다면 박스의 인덱스를 +1 하여 다음 박스로 이동했다.

이걸 반복하면 현재 박스의 남아있는 용량을 체크하여 sum 함수를 통해 정답을 출력할 수 있다.

 

 

 

 

 

 

<코드>

N, M = map(int, input().split())
A = list(map(int, input().split()))
B = list(map(int, input().split()))

i, j = 0, 0  # i=박스 인덱스, j=책 인덱스
while j < M:
    if A[i] >= B[j]:
        A[i] -= B[j]
        j += 1
    else:
        i += 1

print(sum(A))