Computer Science/알고리즘

[프로그래머스 - Python] 42583 - 다리를 지나는 트럭

바보1 2023. 10. 15. 15:37

0.  문제 링크

 

 

https://school.programmers.co.kr/learn/courses/30/lessons/42583

 

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr


1.  풀이 방법

 

 

  1. 이 문제는 deque를 사용하여, 실제 다리와 트럭의 움직임을 구현하면 된다.
  2. 만약 새로운 트럭이 올라갈 수 없는 상태라면. 0을 append 함으로써 빈 공간을 표현한다.

2.  코드

 

 

from collections import deque

def solution(bridge_length, weight, truck_weights):
    answer = 0
    
    dq = deque([0] * bridge_length)
    onBriW = 0
        
    for i in truck_weights:     # 트럭이 정해진 순대로 지나감.
        onBriW -= dq.popleft()      # 맨 앞에 있는 애는 나오고, 다리 위 무게는 줄어듦
        
        while (onBriW + i) > weight:        # 새로운 트럭이 올라갈 수 있는가?
            # 올라갈 수 없으면, 앞에 있는 애들을 빼고 0을 집어넣음.
            answer += 1
            dq.append(0)
            onBriW -= dq.popleft()
        
        dq.append(i)        # 새로운 트럭이 올라감
        onBriW += i     # 무게 추가
        answer += 1     # 트럭 올라가는 시간 추가
    
    return answer + bridge_length

3.  마무리