on_bridge
을 만들어 줬다.
def solution(bridge_length, weight, truck_weights):
answer = 0
on_bridge=[]
# 대기 트럭이 다 pop()될 때까지 while문 반복
while truck_weights:
# 맨 앞에 있는 트럭 무게 가져오기
tmp=truck_weights[0]
answer+=1
# 다리에 트럭이 다 찼을 경우 pop()
if len(on_bridge)>=bridge_length:
on_bridge.pop(0)
# 지금 다리에 트럭이 들어가도 되는 무게인지 비교
if sum(on_bridge)+tmp<=weight:
on_bridge.append(truck_weights.pop(0))
else:
on_bridge.append(0)
# 마지막 트럭이 지나가는 시간을 추가로 넣어줌 (+ bridge_length)
return answer+bridge_length
또.최.못 (또 최적화 못함) ㅠ
한번 최적화하고 코드를 다시 올려보도록 하겠습니다.