알고리즘/코딩 테스트

프로그래머스 javascript 2단계 - 다리를 지나는 트럭

Judith Hopps 2023. 2. 9. 15:43
반응형

다리를 지나는 트럭

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

 

프로그래머스

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

programmers.co.kr

문제 설명

 

트럭 여러 대가 강을 가로지르는 일차선 다리를 정해진 순으로 건너려 합니다. 모든 트럭이 다리를 건너려면 최소 몇 초가 걸리는지 알아내야 합니다. 다리에는 트럭이 최대 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 함수를 완성하세요

문제풀이

- 1.queue 즉 선입선출 구조를 생각해서 풀어야 한다.

- 2. 처음 설정에 1초를 추가하고 queue에 첫 번째 트럭을 넣는다.

- 3. for 문을 돌면서 queue의 무게를 재고 무게가 넘으면 0을 넘지 않으면 트럭을 넣는다.

- 4. 마지막으로 모두다 queue에 들어갔을 때 길의 길이만큼 time을 더해준다. (최종 트럭이 나가는 시간)

 

처음에 0으로 초기화 하지 않고 undefined로 냅뒀더니 시간초과가 되었다.

즉, arr에 undefined로 변수를 저장하지 않고 0으로 데이터를 초기화 하는 것이 좋다.

 

function solution(bridge_length, weight, truck_weights) {
    if (truck_weights.length===1) return bridge_length+1
    let queue = Array(bridge_length).fill(0)
    queue[bridge_length-1] = truck_weights.shift()
    let time = 1
    let sum = queue[bridge_length-1]
    
    while(truck_weights.length){
        time++;
        queue.shift();
        
        // sum -= stack[0] || 0
        
        sum = queue.filter(v=>v).reduce((a,b)=>{return a+b},0)
        // console.log('sum : '+sum+'time : '+time)
        // console.log('stack :' + stack)
        if(sum+truck_weights[0]<=weight){
            queue.push(truck_weights.shift())
            // console.log(truck_weights)
            if (truck_weights.length===0){
                time += bridge_length; break;
            }
        } else{
            queue.push(0)
        }
        
        // console.log('stack')
        // console.log(stack)
    }
    return time
}
반응형