프로그래머스 다리를 지나는 트럭 Swift

문제 설명

트럭 여러 대가 강을 가로지르는 일 차선 다리를 정해진 순으로 건너려 합니다. 모든 트럭이 다리를 건너려면 최소 몇 초가 걸리는지 알아내야 합니다. 트럭은 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
100	100	[10]	101
100	100	[10,10,10,10,10,10,10,10,10,10]	110
2	10	[7,4,5,6]	8

풀이:

 

이번 문제는 스택/큐 문제입니다.

 

문제의 핵심은 제가 생각했을 때 트럭이 올라탈 때의 시간을 기억하는 것인데요.

 

먼저 시간을 잴 변수와  -> var time = 0

 

대기중인 트럭을 나타낼 변수 -> var wait = truck_weights

 

무게를 잴 변수 -> var times = [Int]()

 

각 트럭이 내릴 때 시간을 담는 변수 -> var times = [Int]()

 

트럭의 무게를 기억할 변수 -> var weights = [Int]()

 

이번 문제는 while문을 사용해서 wait 즉 대기 중인 트럭이 없을 때까지 반복합니다.  ->while !wait.isEmpty{...}

 

그리고 시간을 반복이 될 때마다 1씩 증가시켜줍니다. ->  time += 1

 

그 다음 시간이 비어있지 않다면 즉 올라탄 트럭이 있다면 (왜 올라탄 트럭인지는 뒤에 설명) ->  if !times.isEmpty{...)

 

비어있지 않고 만약 현재 시간이 times(트럭이 내릴 때의 시간)과 같다면 -> if time == times.first! {...}

 

times안에 맨 첫번째를 삭제해주고 w에서 트럭의 무게를 담은 것 (weights)중 맨 첫번째만큼 빼주고 트럭의 무게를 담은 것(weights) 중 맨 첫번째를 삭제해줍니다. -> times.removeFirst() w -= weights.first! weights.removeFirst()

 

그 다음 만약 무게가 대기 중인 트럭이 들어와도 버틸 수 있는 무게보다 작거나 같으면 ->  if w+wait.first! <= weight{...}

 

w에 대기중인 트럭의 맨 첫번째 무게를 더해주고 ->  _> w += wait.first!

 

weights배열에 대기중인 트럭의 맨 첫번째 무게를 넣어주고 -> weights.append(wait.first!)

 

times에 현재시간에 다리길이를 합친 (이게 바로 트럭이 내릴 시간입니다. ) 넣어줍니다. -> times.append(time+bridge_length)

 

그 다음 대기열에서 맨 첫번째 트럭을 삭제해줍니다. -> wait.removeFirst()

 

이렇게 wait가 모두 다 없어질 때까지 반복하면 times엔 마지막으로 올라탄 트럭이 내릴 시간만 남게 됩니다. 즉 모든 트럭이 내린 뒤 시간입니다. 고로 시간을 담은 times 배열의 맨 마지막번 째를 반환해줍니다. -> return times.last!

 

 

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27

import Foundation   func solution(_ bridge_length:Int, _ weight:Int, _ truck_weights:[Int]) -> Int {     var time = 0     var wait = truck_weights     var w = 0     var times = [Int]()     var weights = [Int]()          while !wait.isEmpty{         time += 1         if !times.isEmpty{             if time == times.first! {             times.removeFirst()             w -= weights.first!             weights.removeFirst()         }     }         if w+wait.first! <= weight{             w += wait.first!             weights.append(wait.first!)             times.append(time+bridge_length)             wait.removeFirst()         }     }     return times.last! } Colored by Color Scripter