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

문제 주소

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

프로그래머스

문제 설명

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

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

과제 계획을 담은 이차원 문자열 배열 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으로 시작하지 않습니다.
    • 배열은 시간순으로 정렬되어 있지 않을 수 있습니다.
• 진행중이던 과제가 끝나는 시각과 새로운 과제를 시작해야하는 시각이 같은 경우 진행중이던 과제는 끝난 것으로 판단합니다.

---

입출력 예

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];
        
        PriorityQueue<Assignment> pq = new PriorityQueue<>((o1,o2)->o1.time-o2.time);
        PriorityQueue<Assignment> remained = new PriorityQueue<>((o1,o2)->o2.time-o1.time);
        
        for(int i=0; i<plans.length; i++){
            pq.add(new Assignment(toInt(plans[i][1]),plans[i][0],Integer.parseInt(plans[i][2])));
        }
        
        int index = 0;
        Assignment now = pq.poll();
        boolean endPQ = false;
        while(!endPQ){
            
            if(pq.isEmpty()){
                endPQ = true;
                answer[index++] = now.subject;
                continue;
            }

            Assignment next = pq.poll();
            int timediff = timeDiff(now.time,next.time);
            //int diff = timediff - now.cost;
            
            if(timediff - now.cost<0){//다음 것을 실행시작하는데 이번것의 실행시간이 남았으면
                now.cost = now.cost - timediff;
                remained.add(now);
            }else if(timediff - now.cost==0){
                answer[index++] = now.subject;
            }else{//다 실행하고 다음까지 시간이 비면
                answer[index++] = now.subject;
                int diff = diff = timediff - now.cost;
                
                //전의 것 가져와서 실행함
                while(diff>0){
                    
                    if(!remained.isEmpty()){
                        Assignment last = remained.poll();
                        if(last.cost<=diff){
                            diff -= last.cost;
                            answer[index++] = last.subject;
                        }else{
                            last.cost = last.cost-(diff);
                            remained.add(last);
                            break;
                        }
                    }else{
                        diff = 0;
                    }
                }
            }
            
            now = next;
        }
        
        while(!remained.isEmpty()){
            Assignment a = remained.poll();
            answer[index++] = a.subject;
        }
        
        return answer;
    }
    
    public int timeDiff(int i1,int i2){
       int h1 = i1/100;
        int m1 = i1%100;
        int h2 = i2/100;
        int m2 = i2%100;
        
        int hourdiff = h2-h1;
        int mindiff = m2-m1;
        
        return hourdiff*60+mindiff;
        
    }
    
    public int toInt(String s){
        s = s.replace(":","");
        int temp = Integer.parseInt(s);
        return temp;
    }
}

class Assignment{
    public int time;
    public String subject;
    public int cost;
    
    public Assignment(int t,String s,int c){
        time = t;
        subject = s;
        cost = c;
    }
}

테스트 1 〉	통과 (1.55ms, 76.7MB)
테스트 2 〉	통과 (1.60ms, 72.2MB)
테스트 3 〉	통과 (1.66ms, 79.2MB)
테스트 4 〉	통과 (2.04ms, 75.3MB)
테스트 5 〉	통과 (1.90ms, 74.7MB)
테스트 6 〉	통과 (2.13ms, 69.6MB)
테스트 7 〉	통과 (2.30ms, 74.8MB)
테스트 8 〉	통과 (2.29ms, 82.7MB)
테스트 9 〉	통과 (5.38ms, 83.2MB)
테스트 10 〉	통과 (3.21ms, 72.9MB)
테스트 11 〉	통과 (5.12ms, 80.9MB)
테스트 12 〉	통과 (4.19ms, 81.8MB)
테스트 13 〉	통과 (6.24ms, 92.5MB)
테스트 14 〉	통과 (6.77ms, 75.6MB)
테스트 15 〉	통과 (8.09ms, 82MB)
테스트 16 〉	통과 (1.63ms, 73.2MB)
테스트 17 〉	통과 (1.57ms, 77.6MB)
테스트 18 〉	통과 (1.05ms, 67.9MB)
테스트 19 〉	통과 (1.13ms, 74MB)
테스트 20 〉	통과 (3.04ms, 85.9MB)
테스트 21 〉	통과 (3.01ms, 77.6MB)
테스트 22 〉	통과 (5.15ms, 89.6MB)
테스트 23 〉	통과 (5.09ms, 82.3MB)
테스트 24 〉	통과 (6.78ms, 82.9MB)

 

코드 설명

과제를 두 군데에다 저장한다

1. pq: 모든 과제 → 실행시간이 빠른 순으로 저장된 PriorityQueue(이것은 List로 대체 가능하지만 정렬 따로 해주기 귀찮아서 그냥 PriorityQueue 씀)
2. remained : 실행하다가 다른 것을 실행해야 해서 남은 과제 → PriorityQueue에 실행시간이 느린 순으로 저장함(가장 마지막에 하던 과제를 불러와서 다시 해야 하니까)(위의 것 정렬 후 Stack으로 대체 가능)

기본 로직

• 모든 과제를 pq에 저장해줌
• pq에서 하나씩 꺼내서 과제를 수행함
• 과제가 마쳐지면 answer 배열에 과목명을 저장함

 

기본적인 케이스 분류

• 다음 것을 실행시작하는데 이번것의 실행시간이 남았으면) 이번것을 remained에 넣음
• 다음 것을 실행시작하기 전에 시간이 비면) remained에서 가장 최근에 넣은 것을 뽑아서 시간이 다 갈 때까지 진행함
  • 남은 시간보다 과제 수행 시간이 더 길다면) 과제 수행 시간에서 남은 시간만큼 뺴고 다시 remained에 집어넣음
  • 남은 시간보다 과제 수행 시간이 더 짧다면) 남은 시간에서 과제 수행 시간을 빼고 과제가 마쳐졌으니 answer 배열에 넣음, 남은 시간이 없으니 break함 (여기서 break하지 않고 diff에서 값을 뺀 후 전체 while에 diff가 0과 같거나 작으면 break하게 만드는 조건문을 추가하는 방법도 있다)

추가 케이스 분류

• pq가 다 비었으면) 다음 과제가 없다는 거니까 이번 과제를 다 마침
• pq가 다 비었는데 remained가 남아있는 경우 ) 남은 과제를 시작시간이 느린 순으로 하나씩 끝마치고 answer 배열에 넣음

toInt 메서드 설명

• 시간을 비교해야 하는데 String으로 비교 로직 짜기 귀찮아서 편의상 숫자로 변경해줌

timeDiff 메서드 설명

• 편의상 숫자로 변경해놓은 시간과 분을 분 단위로 바꾸어서 둘의 차를 구해줌
• 앞 두자리가 숫자, 뒤 두자리가 분이라 100단위로 몫과 나머지로 분리해서 진행

Assignment 클래스 설명

• 시간, 과목명, 숙제를 마칠때까지 남은 시간

풀이 과정

List와 Stack으로도 풀 수 있는 문제이지만 미리 List에 값을 다 넣어두고 정렬하는 것보다

PriorityQueue에 넣으면서 정렬되게 하는게 더 빠를 것 같아서 그렇게 풀었다