====== 다리를 지나는 트럭 ====== ===== 풀이 ===== * 다리 위에 있는 트럭들을 큐로 관리해주면 쉽게 처리되는 문제이다. * 그것을 나이브하게 풀면, 매 초마다 새 트럭이 다리에 올라갈수 없는 경우에 무게 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 {{tag>프로그래머스 ps:problems:programmers:Level_2}}