ps:problems:programmers:42583
다리를 지나는 트럭
ps | |
---|---|
링크 | programmers.co.kr/… |
출처 | 프로그래머스 |
문제 번호 | 42583 |
문제명 | 다리를 지나는 트럭 |
레벨 | Level 2 |
분류 |
큐 |
시간복잡도 | O(n) |
인풋사이즈 | n<=10000 |
사용한 언어 | Python |
해결날짜 | 2021/05/21 |
태그 |
풀이
- 다리 위에 있는 트럭들을 큐로 관리해주면 쉽게 처리되는 문제이다.
- 그것을 나이브하게 풀면, 매 초마다 새 트럭이 다리에 올라갈수 없는 경우에 무게 0짜리의 가상 트럭을 대신 다리에 올리는 것처럼 구현할 수 있고, 이 경우에는 전체 걸린 시간, 즉 O(n*m)에 풀린다 (n=트럭의 갯수, m=다리의 길이). 비효율적이지만, 정답에 올라가있는 대부분의 코드가 이러한 방식으로 구현되어있다. (코드 1)
- 트럭마다 트럭의 무게와 다리에 올라간 시간을 함께 묶어서 큐에 저장하면, 가상 트럭 없이 실제 트럭들에 대해서만 큐에 한번씩만 추가/삭제하는 것으로 전체 걸린 시간을 계산할 수 있다. 이렇게 하면 O(n)에 해결 가능하다. (코드2)
코드
코드 1 - O(n*m)
"""Solution code for "Programmers 42583. 다리를 지나는 트럭".
- Problem link: https://programmers.co.kr/learn/courses/30/lessons/42583
- Solution link: http://www.teferi.net/ps/problems/programmers/42583
"""
import collections
def solution(bridge_length, weight, truck_weights):
weight_sum = 0
trucks_on_bridge = collections.deque([0] * bridge_length)
sec = 0
for truck_weight in truck_weights:
while True:
sec += 1
weight_sum -= trucks_on_bridge.popleft()
if weight_sum + truck_weight <= weight:
trucks_on_bridge.append(truck_weight)
weight_sum += truck_weight
break
trucks_on_bridge.append(0)
return sec + bridge_length
코드 2 - O(n)
"""Solution code for "Programmers 42583. 다리를 지나는 트럭".
- Problem link: https://programmers.co.kr/learn/courses/30/lessons/42583
- Solution link: http://www.teferi.net/ps/problems/programmers/42583
"""
import collections
Truck = collections.namedtuple('Truck', ['weight', 'enter_time'])
def solution(bridge_length, weight, truck_weights):
weight_sum = 0
trucks_on_bridge = collections.deque()
cur_sec = 0
for cur_truck_weight in truck_weights:
cur_sec += 1
if (trucks_on_bridge and
trucks_on_bridge[0].enter_time + bridge_length <= cur_sec):
weight_sum -= trucks_on_bridge.popleft().weight
while weight_sum + cur_truck_weight > weight:
out_truck = trucks_on_bridge.popleft()
cur_sec = out_truck.enter_time + bridge_length
weight_sum -= out_truck.weight
trucks_on_bridge.append(Truck(cur_truck_weight, cur_sec))
weight_sum += cur_truck_weight
return cur_sec + bridge_length
ps/problems/programmers/42583.txt · 마지막으로 수정됨: 2021/05/21 07:16 저자 teferi
토론