[프로그래머스 Level.2] 과제 진행하기 (연습 문제) (Java)
문제 링크
https://school.programmers.co.kr/learn/courses/30/lessons/176962
프로그래머스
코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.
programmers.co.kr
코딩테스트 연습 > 연습문제 > 과제 진행하기
문제 설명
과제를 받은 루는 다음과 같은 순서대로 과제를 하려고 계획을 세웠습니다.
- 과제는 시작하기로 한 시각이 되면 시작합니다.
- 새로운 과제를 시작할 시각이 되었을 때, 기존에 진행 중이던 과제가 있다면 진행 중이던 과제를 멈추고 새로운 과제를 시작합니다.
- 진행중이던 과제를 끝냈을 때, 잠시 멈춘 과제가 있다면, 멈춰둔 과제를 이어서 진행합니다.
- 만약, 과제를 끝낸 시각에 새로 시작해야 되는 과제와 잠시 멈춰둔 과제가 모두 있다면, 새로 시작해야 하는 과제부터 진행합니다.
- 멈춰둔 과제가 여러 개일 경우, 가장 최근에 멈춘 과제부터 시작합니다.
과제 계획을 담은 이차원 문자열 배열 plans가 매개변수로 주어질 때, 과제를 끝낸 순서대로 이름을 배열에 담아 return 하는 solution 함수를 완성해주세요.
제한사항
- 3 ≤ plans의 길이 ≤ 1,000
- plans의 원소는 [name, start, playtime]의 구조로 이루어져 있습니다.
- name : 과제의 이름을 의미합니다.
- 2 ≤ name의 길이 ≤ 10
- name은 알파벳 소문자로만 이루어져 있습니다.
- name이 중복되는 원소는 없습니다.
- start : 과제의 시작 시각을 나타냅니다.
- "hh:mm"의 형태로 "00:00" ~ "23:59" 사이의 시간값만 들어가 있습니다.
- 모든 과제의 시작 시각은 달라서 겹칠 일이 없습니다.
- 과제는 "00:00" ... "23:59" 순으로 시작하면 됩니다. 즉, 시와 분의 값이 작을수록 더 빨리 시작한 과제입니다.
- playtime : 과제를 마치는데 걸리는 시간을 의미하며, 단위는 분입니다.
- 1 ≤ playtime ≤ 100
- playtime은 0으로 시작하지 않습니다.
- 배열은 시간순으로 정렬되어 있지 않을 수 있습니다.
- name : 과제의 이름을 의미합니다.
- plans의 원소는 [name, start, playtime]의 구조로 이루어져 있습니다.
- 진행중이던 과제가 끝나는 시각과 새로운 과제를 시작해야하는 시각이 같은 경우 진행중이던 과제는 끝난 것으로 판단합니다.
입출력 예
plans | result |
[["korean", "11:40", "30"], ["english", "12:10", "20"], ["math", "12:30", "40"]] | ["korean", "english", "math"] |
[["science", "12:40", "50"], ["music", "12:20", "40"], ["history", "14:00", "30"], ["computer", "12:30", "100"]] | ["science", "history", "computer", "music"] |
[["aaa", "12:00", "20"], ["bbb", "12:10", "30"], ["ccc", "12:40", "10"]] | ["bbb", "ccc", "aaa"] |
입출력 예 설명
입출력 예 #1
"korean", "english", "math"순으로 과제를 시작합니다. "korean" 과제를 "11:40"에 시작하여 30분 후인 "12:10"에 마치고, 즉시 "english" 과제를 시작합니다. 20분 후인 "12:30"에 "english" 과제를 마치고, 즉시 "math" 과제를 시작합니다. 40분 후인 "01:10"에 "math" 과제를 마칩니다. 따라서 "korean", "english", "math" 순으로 과제를 끝내므로 차례대로 배열에 담아 반환합니다.
입출력 예 #2
"music", "computer", "science", "history" 순으로 과제를 시작합니다.
시각 | 진행 중 과제 | 잠시 멈춘 과제 | 설명 |
"12:20" | "music" | [ ] | "music"을 시작합니다. |
"12:30" | "computer" | ["music"] | "music"을 잠시 멈추고(남은 시간 30분) "computer"를 시작합니다 |
"12:40" | "science" | ["music", "computer"] | "computer"를 잠시 멈추고(남은 시간 90분) "science"를 시작합니다 |
"13:30" | "computer" | ["music"] | "science"를 끝내고 가장 최근에 멈춘 "computer"를 다시 시작합니다 |
"14:00" | "history" | ["music", "computer"] | "computer"를 잠시 멈추고(남은 시간 60분) "history"를 시작합니다 |
"14:30" | "computer" | ["music"] | "history"를 끝내고 가장 최근에 멈춘 "computer"를 다시 시작합니다" |
"15:30" | "music" | [ ] | "computer"를 끝내고 가장 최근에 멈춘 "music"을 다시 시작합니다" |
"16:00" | - | [ ] | "music"을 끝냅니다 |
따라서 ["science", "history", "computer", "music"] 순서로 과제를 마칩니다.
입출력 예 #3
설명 생략
나의 코드
import java.util.*;
class Solution {
public String[] solution(String[][] plans) {
String[] answer = new String[plans.length];
Plan[] planArr = new Plan[plans.length];
int resultIdx = 0;
for(int i=0; i<plans.length; i++) {
String[] starts = plans[i][1].split(":");
int start = Integer.parseInt(starts[0]) * 60 + Integer.parseInt(starts[1]);
planArr[i] = new Plan(plans[i][0], start, Integer.parseInt(plans[i][2]));
}
Arrays.sort(planArr, (o1, o2) -> (o1.start - o2.start));
Stack<Plan> stack = new Stack<>();
stack.push(planArr[0]);
int time = planArr[0].start;
int planIdx = 1;
while(!stack.isEmpty() || planIdx < planArr.length) {
if(planIdx >= planArr.length) {
while(!stack.isEmpty()) {
Plan beforePlan = stack.pop();
time += beforePlan.playtime;
answer[resultIdx++] = beforePlan.name;
}
break;
}
while(!stack.isEmpty() && time + stack.peek().playtime <= planArr[planIdx].start) {
Plan beforePlan = stack.pop();
time += beforePlan.playtime;
answer[resultIdx++] = beforePlan.name;
}
if(!stack.isEmpty()) {
Plan beforePlan = stack.pop();
beforePlan.playtime -= planArr[planIdx].start - time;
stack.push(beforePlan);
}
time = planArr[planIdx].start;
stack.push(planArr[planIdx++]);
}
return answer;
}
class Plan {
String name;
int start;
int playtime;
Plan(String name, int start, int playtime) {
this.name = name;
this.start = start;
this.playtime = playtime;
}
}
}
풀이
- 과제를 진행하다 다음 과제 시작 시간이 되면 해당 과제를 멈추고 다음 과제를 시작하고 다시 멈추었던 최근 과제를 꺼내 진행해야 하므로 과제를 시작 시간으로 오름차순으로 정렬하고 Stack을 이용해 최근 과제를 꺼내 다시 진행하도록 한다.
- 우선 과제의 name, start, playtime을 멤버 변수로 하는 Plan 클래스를 생성하고 Plan 배열 planArr을 생성한다.
- 주어진 plans를 순회하면서 start 시간을 분으로 변환하고 start 시간과 playtime을 int형으로 변환하여 담는다.
- Plan 배열을 start를 기준으로 오름차순 정렬한다.
- Plan을 선언 타입으로 하는 Stack을 생성하고 첫 과제를 Stack에 담는다.
- 시간을 나타낼 int형 변수 time을 선언하고 첫 과제의 start로 초기화한다.
- planArr 배열의 인덱스를 나타낼 planIdx를 선언하고 첫 과제는 Stack에 담았으므로 planIdx를 1로 초기화한다.
- 이제 스택이 비어있고 planIdx도 planArr 배열의 길이와 같아지면 더 이상 할 과제가 없으므로 종료시키는 while문을 생성한다.
- planIdx가 planArr의 길이와 같아지면 더 이상 다음 과제가 없으므로 멈추었던 과제들만 전부 마무리하고 종료하면 된다. 따라서 planIdx가 planArr 길이 이상이면 스택이 빌 때까지 스택에 담아놓은 과제들을 차례로(최근 과제들부터) 꺼내면서 해당 과제의 name을 answer에 차례로 저장하고 종료한다.
- 그게 아니라면 스택이 비어있지 않고 time(현재 시간) + 스택의 최근 과제의 playtime이 다음 과제의 시작 시간보다 같거나 작다면 최근 과제를 꺼내서 해당 과제의 playtime만큼 time에 더해주고 해당 과제를 마무리했으니 answer에 해당과제의 name을 저장한다. 해당 작업을 while문을 통해 반복하도록 한다. (스택에 멈춘 과제들이 여러 개 담겨있고 다음 과제 시작시간까지 시간이 많이 남아있으면 계속해서 멈췄던 과제들을 차례로 꺼내서 하면 된다)
- 이제 while문을 종료하고 나왔을 때 스택이 비어있지 않다면 time + 최근 과제의 playtime이 다음 시작 과제의 시작 시간보다 크다는 뜻이므로 최근 과제를 끝내지 못한다는 것이다. 하지만 끝내지 못하더라도 다음 과제 시작 전까지는 해당 과제를 진행할 수 있으므로 시작 전까지는 최근 과제를 진행하고 해당 진행 시간만큼 playtime을 줄여줘야 한다. 따라서 최근 과제를 꺼내서 최근 과제의 playtime에서 다음 과제 시작 시간 - 현재 시간만큼을 빼준 후에 다시 스택에 담는다.
- 그리고 나서 time을 다음 과제 시작 시간으로 변경하고 스택에 다음 과제를 담아준 후에 planIdx를 1 증가시키면 된다.
- 이렇게 while문을 반복한 후에 종료하고 나오면 answer에는 종료시킨 과제 순서대로 저장되어있으므로 answer를 그대로 반환하면 된다.