Problem Solving/Programmers

【프로그래머스】 스택/큐 - 기능개발 (Kotlin)

까망사과 2022. 7. 1. 13:00

문제 설명

프로그래머스 팀에서는 기능 개선 작업을 수행 중입니다. 각 기능은 진도가 100%일 때 서비스에 반영할 수 있습니다.

또, 각 기능의 개발속도는 모두 다르기 때문에 뒤에 있는 기능이 앞에 있는 기능보다 먼저 개발될 수 있고, 이때 뒤에 있는 기능은 앞에 있는 기능이 배포될 때 함께 배포됩니다.

 

먼저 배포되어야 하는 순서대로 작업의 진도가 적힌 정수 배열 progresses와 각 작업의 개발 속도가 적힌 정수 배열 speeds가 주어질 때 각 배포마다 몇 개의 기능이 배포되는지를 return 하도록 solution 함수를 완성하세요.

 

제한 사항

  • 작업의 개수(progresses, speeds 배열의 길이)는 100개 이하입니다.
  • 작업 진도는 100 미만의 자연수입니다.
  • 작업 속도는 100 이하의 자연수입니다.
  • 배포는 하루에 한 번만 할 수 있으며, 하루의 끝에 이루어진다고 가정합니다.
    예를 들어 진도율이 95%인 작업의 개발 속도가 하루에 4%라면 배포는 2일 뒤에 이루어집니다.

 

입출력 예

progresses speeds return
[93, 30, 55] [1, 30, 5] [2, 1]
[95, 90, 99, 99, 80, 99] [1, 1, 1, 1, 1, 1] [1, 3, 2]

 

입출력 예 설명

입출력 예 #1
첫 번째 기능은 93% 완료되어 있고 하루에 1%씩 작업이 가능하므로 7일간 작업 후 배포가 가능합니다.
두 번째 기능은 30%가 완료되어 있고 하루에 30%씩 작업이 가능하므로 3일간 작업 후 배포가 가능합니다.

하지만 이전 첫 번째 기능이 아직 완성된 상태가 아니기 때문에 첫 번째 기능이 배포되는 7일째 배포됩니다.
세 번째 기능은 55%가 완료되어 있고 하루에 5%씩 작업이 가능하므로 9일간 작업 후 배포가 가능합니다.

따라서 7일째에 2개의 기능, 9일째에 1개의 기능이 배포됩니다.

 

입출력 예 #2
모든 기능이 하루에 1%씩 작업이 가능하므로, 작업이 끝나기까지 남은 일수는 각각 5일, 10일, 1일, 1일, 20일, 1일입니다.

어떤 기능이 먼저 완성되었더라도 앞에 있는 모든 기능이 완성되지 않으면 배포가 불가능합니다.

따라서 5일째에 1개의 기능, 10일째에 3개의 기능, 20일째에 2개의 기능이 배포됩니다.

 


풀이

각 기능이 배포될 때까지 걸리는 일수를 큐에 저장한 다음, 배포마다 배포 가능한 기능을 제거하는 방법을 사용했다.

 

먼저 배포마다 배포되는 기능 개수를 저장하기 위한 정답 배열, 그리고 소요 일수를 저장하기 위한 큐가 필요하다.

정답 배열은 크기를 예측하기 어려우므로 크기를 변경할 수 있는 ArrayList를 사용하고 큐는 LinkedList를 사용했다.

import java.util.LinkedList

val answer = arrayListOf<Int>()
val queue = LinkedList<Int>()

 

기능마다 배포까지 걸리는 일수를 계산해야 한다.

(100 - 현재 진행률) / 작업 속도를 계산한 값을 올림해주면 된다.

 

올림하는데 kotlin.math 라이브러리의 ceil() 함수를 사용할 수 있지만, 형변환이 필요하기도 하고 import를 되도록 덜 하고 싶어서 나머지를 이용하여 아래와 같이 값을 계산했다.

 

계산한 일수는 순서대로 큐에 추가한다.

for (i in progresses.indices) {
    val remaining = 100 - progresses[i]
    val speed = speeds[i]
    
    queue.add(remaining / speed + if (remaining % speed == 0) 0 else 1)
}

 

반복문을 돌면서 배포마다 배포 가능한 기능을 큐에서 제거하고, 개수를 세서 정답 배열에 추가한다.

앞에 있는 기능이 배포되어야 그 뒤의 기능도 배포 가능하므로, 배포마다 맨 앞 기능의 소요 일수보다 큰 값을 만날 때까지 카운트를 늘린다.

그러고 나서 이 카운트만큼 큐의 맨 앞 요소를 제거한다.

 

예를 들어 큐가 [7, 3, 6, 9]라 하면 처음 배포는 7일차 때 진행될 것이다.

2번째 기능은 3일차부터, 3번째 기능은 6일차부터 배포할 수 있으므로 1번째 기능과 함께 배포할 수 있다.

하지만 4번째 기능은 배포되려면 2일 더 남았기 때문에 여기에서 배포를 멈춘다.

정답 배열은 [3, 1]이 된다.

 

위 작업을 모든 기능이 배포될 때까지, 다시 말하자면 큐가 공백 상태가 될 때까지 반복한다.

while (queue.isNotEmpty()) {
    var count = 0
    val first = queue.peek()
    
    for (day in queue) {
        if (first < day) break
        else count++
    }

    repeat(count) { queue.remove() }
    answer.add(count)
}

 

이렇게 정답 배열이 완성되었고 IntArray 타입으로 형변환을 한 뒤 반환하면 된다.

 

코드

import java.util.LinkedList

fun solution(progresses: IntArray, speeds: IntArray): IntArray {
    val answer = arrayListOf<Int>()
    val queue = LinkedList<Int>()

    // 각 기능이 배포될 때까지 걸리는 일수를 큐에 추가
    for (i in progresses.indices) {
        val remaining = 100 - progresses[i]
        val speed = speeds[i]

        queue.add(remaining / speed + if (remaining % speed == 0) 0 else 1)
    }

    // 배포할 수 있는 기능을 한꺼번에 큐에서 제거하고 개수를 정답 배열에 추가
    while (queue.isNotEmpty()) {
        var count = 0
        val first = queue.peek()

        for (day in queue) {
            if (first < day) break
            else count++
        }

        repeat(count) { queue.remove() }
        answer.add(count)
    }

    return answer.toIntArray()
}