본문 바로가기
  • 문과생의 백엔드 개발자 성장기
|Algorithm

[1주차]_프로그래머스 Lv2 다리를 지나는 트럭(스택/큐) . PYTHON

by 케리's 2021. 5. 23.

다리를 지나는 트럭

 

문제 설명

 

트럭 여러 대가 강을 가로지르는 일 차선 다리를 정해진 순으로 건너려 합니다.

모든 트럭이 다리를 건너려면 최소 몇 초가 걸리는지 알아내야 합니다.

트럭은 1초에 1만큼 움직이며, 다리 길이는 bridge_length이고 다리는 무게 weight까지 견딥니다.


※ 트럭이 다리에 완전히 오르지 않은 경우, 이 트럭의 무게는 고려하지 않습니다.

예를 들어, 길이가 2이고 10kg 무게를 견디는 다리가 있습니다. 무게가 [7, 4, 5, 6]kg인 트럭이 순서대로 최단 시간 안에 다리를 건너려면 다음과 같이 건너야 합니다.

 

 

경과 시간                   다리를 지난 트럭                       다리를 건너는 트럭      대기 트럭

0 [] [] [7,4,5,6]
1~2 [] [7] [4,5,6]
3 [7] [4] [5,6]
4 [7] [4,5] [6]
5 [7,4] [5] [6]
6~7 [7,4,5] [6] []
8 [7,4,5,6] [] []

따라서, 모든 트럭이 다리를 지나려면 최소 8초가 걸립니다.

solution 함수의 매개변수로 다리 길이 bridge_length, 다리가 견딜 수 있는 무게 weight, 트럭별 무게 truck_weights가 주어집니다. 이때 모든 트럭이 다리를 건너려면 최소 몇 초가 걸리는지 return 하도록 solution 함수를 완성하세요.

제한 조건

  • bridge_length는 1 이상 10,000 이하입니다.
  • weight는 1 이상 10,000 이하입니다.
  • truck_weights의 길이는 1 이상 10,000 이하입니다.
  • 모든 트럭의 무게는 1 이상 weight 이하입니다.

 

입출력 예

 

bridge_length   weight           truck_weights                                                                              return

2 10 [7,4,5,6] 8
100 100 [10] 101
100 100 [10,10,10,10,10,10,10,10,10,10] 110

 

 

✔ 나의 문제 설명

 

시간은 반복 할때 마다 1초씩 늘어난다.

다리를 지난 트럭 리스트, 다리를 건너는 트럭 리스트가 필요하다.

대기트럭은 트럭이 다리에 올라갈 때마다 가장 첫 번째 원소가 빠진다. = queue 활용

 

그럼 트럭이 다리에 올라갈 수 있는 지 판단하는 조건은

현재 다리에 올라와 있는 트럭의 무게에 트럭의 원소 첫 번째 무게를 합한 값이 weight보다 작으면 트럭이 올라감

 ex) 길이가 2인 다리, 무게가 10kg까지 견딤, 7kg트럭이 다리위에있으면 10>7 트럭올라갈 수 있다.

 

그렇지 않으면 트럭은 올라가지 못한다.

 

 

😎 나의 풀이

 

def solution(bridge_length, weight, truck_weights):
    # 1. 초를 세주는 변수, 반복할때 마다 1씩 증가
    answer = 0
    # 2. 다리 길이만큼 리스트 만들자, 트럭이 시간에 따라 지날때 마다 반영 한다.
    bridge_1 = [0] * bridge_length
    # 3. 현재 다리위에 올라와있는 트럭의 무게, 0으로 초기화 하자.
    weight_1 = 0
    #4. 트럭이 남아있을 때 까지 반복하자.
    while truck_weights:
        # 5. 초를 한개씩 늘려주자
        answer += 1
        # 6. bridege_2 라는 객체에 bridge_1 의 첫번째 객체 저장하자, 다음음에빠질 트럭의 무게의미한다.
        bridge_2 = bridge_1.pop(0)
        # 7. whight_1 에는 bridge_2 을 뺀값을 저장하자.
        weight_1 -= bridge_2
        # 8. 현재트럭 무게 weigth_1 와 + truck_weights의 첫번째 합이 > 최대 수용할수 있는 weight 중량보다 작다면
        if weight_1 + truck_weights[0] > weight:
            bridge_1.append(0)
        # 9. truck에 truck_weights 첫번째 값을 뺀다
        else:
            truck = truck_weights.pop(0)
        #10. bridge_1에 truck을 추가한다. 이후 weight_1 에도 truck무게를 추가한다.
            bridge_1.append(truck)
            weight_1 += truck

    while weight_1 > 0:
        answer += 1
        bridge_2 = bridge_1.pop(0)
        weight_1 -= bridge_2

    return answer

댓글