[프로그래머스 LV2] 기능개발 (Queue, 큐)

🧐 프로그래머스 LV2. 기능개발

문제 설명

프로그래머스 팀에서는 기능 개선 작업을 수행 중입니다. 각 기능은 진도가 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]

https://school.programmers.co.kr/learn/courses/30/lessons/42586?language=java

프로그래머스

 

💡 문제 풀이

단순 반복문에 큐를 조금 사용하면 금방 풀리는 문제였다.

큐에 progress와 speed 정보를 추가하고, 진행률이 완료되면 차례대로 배포된 작업의 개수를 계산해서 answer 배열에 추가하면 된다.

 

progress와 speed 정보를 한 번에 저장하기 위해서 클래스를 정의해주었다.

static class Work {
    int progress;
    int speed;
        
    Work(int progress, int speed) {
        this.progress = progress;
        this.speed = speed;
    }
        
    public int getTotalProgress(int day) { // 특정 일 수(day)가 지난 후 진행률
        return this.progress + this.speed * day;
    }
}

진행률인 progress와 개발 속도인 speed 정보를 저장하고 있고,

getTotalProgress 메소드에서 특정 일 수(day)가 지난 후의 진행률을 구할 수 있다.

 

progresses와 speeds의 정보를 큐에 저장한다.

Queue<Work> queue = new LinkedList<>();
for (int i = 0; i < progresses.length; i++) {
    queue.add(new Work(progresses[i], speeds[i]));
}

 

이제 반복문을 돌면서 조건에 따라 작업 클래스 객체를 처리하고, 결과를 구한다.

List<Integer> al = new ArrayList<>(); // 각 배포의 작업 개수를 포함하는 리스트
int day = 0, count = 0; // day: 지난 일 수, count: 배포 대기 작업 개수
        
while (!queue.isEmpty()) {
    Work work = queue.peek(); // 남은 작업 중 우선순위가 가장 높은 작업
            
    if (work.getTotalProgress(day) >= 100) { // 진행 완료(배포)
        queue.poll();
        count++; // 배포 개수 증가
    } else { // 진행 중
        if (count > 0) { // 배포 작업이 있으면 반영 후 초기화
            al.add(count);
            count = 0;
        }
        day++; // 일 수 증가
    }
}
        
if (count > 0) al.add(count); // 배포된 작업이 있으면 마지막 반영

우선순위가 높은 작업 순서대로 배포 여부를 판단하고 처리한다.

배포가 가능하다면, 큐에서 꺼내고 배포 개수를 증가시킨다.

배포 전이라면, 배포 한 작업이 있다면(count > 0) 결과 리스트에 반영해주고 값을 0으로 초기화한다. 그리고 일 수를 증가시킨다.

반복문이 끝난 후, 배포한 작업이 있다면 마지막으로 결과 리스트에 반영해준다.

 

위 풀이 방식을 전체 코드로 풀어내면 다음과 같다.

import java.util.*;

class Solution {
    
    static class Work {
        int progress;
        int speed;
        
        Work(int progress, int speed) {
            this.progress = progress;
            this.speed = speed;
        }
        
        public int getTotalProgress(int day) {
            return this.progress + this.speed * day;
        }
    }
    
    public int[] solution(int[] progresses, int[] speeds) {
        Queue<Work> queue = new LinkedList<>();
        for (int i = 0; i < progresses.length; i++) {
            queue.add(new Work(progresses[i], speeds[i]));
        }
        
        List<Integer> al = new ArrayList<>();
        int day = 0, count = 0;
        
        while (!queue.isEmpty()) {
            Work work = queue.peek();
            
            if (work.getTotalProgress(day) >= 100) { // 진행 완료
                queue.poll();
                count++;
            } else { // 진행 중
                if (count > 0) { // 배포 작업이 있으면 반영 후 초기화
                    al.add(count);
                    count = 0;
                }
                day++;
            }
        }
        
        if (count > 0) al.add(count);
        
        int[] answer = new int[al.size()];
        for (int i = 0; i < al.size(); i++) {
            answer[i] = al.get(i);
        }
        
        return answer;
    }
}

 

✨ 포스팅을 마치며

이번 문제도 조건와 처리 방법만 잘 이해하고 정리하면 금방 쉽게 풀 수 있었다. 코딩테스트에서는 제한된 시간이 있으므로 문제를 빠르게 읽고 이해하는 것에 익숙해는 것이 중요하다고 생각되었다.