[프로그래머스] 과제 진행하기 - Java

문제 링크

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

문제 설명

과제를 받은 루는 다음과 같은 순서대로 과제를 하려고 계획을 세웠습니다.

• 과제는 시작하기로 한 시각이 되면 시작합니다.

• 새로운 과제를 시작할 시각이 되었을 때, 기존에 진행 중이던 과제가 있다면 진행 중이던 과제를 멈추고 새로운 과제를 시작합니다.

• 진행중이던 과제를 끝냈을 때, 잠시 멈춘 과제가 있다면, 멈춰둔 과제를 이어서 진행합니다.
  만약, 과제를 끝낸 시각에 새로 시작해야 되는 과제와 잠시 멈춰둔 과제가 모두 있다면, 새로 시작해야 하는 과제부터 진행합니다.
  멈춰둔 과제가 여러 개일 경우, 가장 최근에 멈춘 과제부터 시작합니다.

과제 계획을 담은 이차원 문자열 배열 plans가 매개변수로 주어질 때, 과제를 끝낸 순서대로 이름을 배열에 담아 return 하는 solution 함수를 완성해주세요.

제한 사항

입출력 예

알고리즘: 정렬, 스텍, 구현

풀이

과제는 시작하기로 한 시각이 되면 시작하기 때문에, 시작 시각을 기준으로 오름차순 정렬해준다. -> 정렬

시작 시각과 남은 시간은 문자열이기 때문에 비교에 용이하게끔 정수로 바꾸고 단위는 분 단위로 바꿔준다. -> 구현

진행중인 과제를 다음 과제 시작 전에 못 끝냈다면 stack에 넣어준다. stack에 넣는 이유는 멈춰둔 과제가 여러 개일 경우 가장 최근에 멈춘 과제부터 시작해야 하므로 LIFO 구조가 최적이다. -> 스택

이제 문제 조건대로 구현해주면 된다. -> 구현

소스 코드

import java.util.*;
class Plan {
    String s;
    int st, lt;
    Plan(String s, int st, int lt) {
        this.s = s;
        this.st = st;
        this.lt = lt;
    }
}
class Solution {
    static Plan[] ob_plans;
    static Stack<Plan> wait_stack = new Stack<>();
    static ArrayList<String> end_list = new ArrayList<>();
    public String[] solution(String[][] plans) {
        String[] answer = new String[plans.length];
        ob_plans = new Plan[plans.length];
        for(int i=0; i<plans.length; i++) {
            StringTokenizer st = new StringTokenizer(plans[i][1], ":");
            int min = Integer.parseInt(st.nextToken()) * 60 + Integer.parseInt(st.nextToken());
            ob_plans[i] = new Plan(plans[i][0], min, Integer.parseInt(plans[i][2]));
        }
        Arrays.sort(ob_plans, (o1, o2) -> {
           return o1.st - o2.st; 
        });
        
        for(int i=0; i<ob_plans.length; i++) {
            if(i == ob_plans.length - 1) {
                end_list.add(ob_plans[i].s);
                while(wait_stack.size()!=0) {
                    end_list.add(wait_stack.pop().s);
                }
            } else {
                int left = ob_plans[i+1].st - (ob_plans[i].st + ob_plans[i].lt);
                if(left >= 0) {
                    end_list.add(ob_plans[i].s);
                    while(wait_stack.size()!=0 && left != 0) {
                        Plan top = wait_stack.pop();
                        if(top.lt<=left) {
                            end_list.add(top.s);
                            left -= top.lt;
                        } else {
                            top.lt -= left;
                            left = 0;
                            wait_stack.push(top);
                        }
                    }
                } else wait_stack.push(new Plan(ob_plans[i].s, ob_plans[i].st, Math.abs(left)));
            }
        }
        for(int i=0; i<end_list.size(); i++) answer[i] = end_list.get(i);
        return answer;
    }
}