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

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

프로그래머스

 

 

 

 

 

문제

• 과제 계획을 담은 이차원 문자열 배열 plans가 매개변수로 주어질 때, 과제를 끝낸 순서대로 이름을 배열에 담아 return
  • plans의 원소는 [name, start, playtime]의 구조
  • name : name이 중복되는 원소는 없음
  • start : "hh:mm"의 형태
  • playtime : 단위는 분
• 새로운 과제를 시작할 시각이 되었을 때, 진행 중이던 과제를 멈추고 새로운 과제를 시작
• 과제를 끝냈을 때, 멈춰둔 과제를 이어서 진행
  • 만약, 과제를 끝낸 시각에 새로 시작해야 되는 과제와 잠시 멈춰둔 과제가 모두 있다면, 새로 시작해야 하는 과제부터 진행
• 멈춰둔 과제가 여러 개일 경우, 가장 최근에 멈춘 과제부터 시작

 

입출력 예시

• Plans : [["science", "12:40", "50"], ["music", "12:20", "40"], ["history", "14:00", "30"], ["computer", "12:30", "100"]]
• result : ["science", "history", "computer", "music"]

 

 

 

 

 

풀이

• class (name, start, playtime) 만들어서 리스트에 저장
  • start : 계산하기 쉽게 파싱해서 분(int)으로 바꾸기
• 시작순으로 오름차순 정렬 후 시작
• 가장 최근에 멈춘 과제부터 시작해야 하므로 진행중 과제를 Stack에 저장
  
  • 스택에 진행해야 하는 과제 넣고 시작
  • 다음 과제 시작 전까지 스택에서 과제 빼서 마침
  • 과제 시작 시간이 되면, 스택에 못 마친 과제의 playtime을 남은 시간으로 갱신해서 넣음 -> 그리고 중단
  • 위 과정을 반복하면 제 때 시작해야 하는 과제는 끝남
  • 스택에 남아있는 잔여 과제들을 차례로 수행
• 결과는 List에 저장 후, return 시 배열로 변환

 

 

 

스택 부분에서 문제에서 정해진 과정에 따라 처리하는 것이 어려웠다.

 

 

 

 

 

코드

import java.util.*;

class Solution {
    public class Plan {
        String name;
        int start, playtime;
        
        public Plan(String name, int start, int playtime) {
            this.name = name;
            this.start = start;
            this.playtime = playtime;
        }
        
        public int getStart() {
            return start;
        }
    }
    
    
    public String[] solution(String[][] plans) {
        List<Plan> planList = new ArrayList<>(); //Plan 클래스로 만들어 리스트로 저장
        
        for(String[] plan : plans) {
            planList.add(new Plan(plan[0], startToMin(plan[1]), Integer.parseInt(plan[2]) ));
        }
        
        //시작순으로 오름차순 정렬 후 시작
        Collections.sort(planList, (p1, p2) -> p1.getStart() - p2.getStart());
        
        Stack<Plan> stack = new Stack<>(); //과제 대기열
        List<String> result = new ArrayList<>(); //결과
        
        for(int i = 0; i < planList.size(); i++) {
            int now = planList.get(i).start;
            int next = 0;
            if(i != planList.size() - 1)  {
                next = planList.get(i + 1).start;
            }
            
            stack.add(planList.get(i));
            
            while(!stack.isEmpty()) {
                Plan p = stack.pop();
                
                if(now + p.playtime <= next) {
                    now += p.playtime; //현재 시각 갱신
                    result.add(p.name); //과제 마침
                } else { //과재 못마침
                    stack.add(new Plan(p.name, p.start, now + p.playtime - next)); //중단하고 과제 대기열
                    // now = next;
                    break;
                }
            }
        }
        
        //잠시 멈춘 과제 모두 다시 시작
        while (!stack.isEmpty()) {
            result.add(stack.pop().name);
        }
        
        //List -> 배열 변환
        return result.toArray(new String[0]);
    }
    
    public int startToMin(String start) { //시작시간을 분으로 변환
        String[] time = start.split(":");
        
        return Integer.parseInt(time[0]) * 60 + Integer.parseInt(time[1]);
    }
}

 

 

 

 

 

결과

 

 

 

 

배운점

• 객체 리스트 오름차순 정렬

Collections.sort(planList, (p1, p2) -> p1.getStart() - p2.getStart());

 

• List -> 배열 변환

result.toArray(new String[0])

 

• 스택으로 정해진 과정을 구현하는 것이 어려웠다. 특히 예외 상황 없이 간단 명료하게 구현하도록 하는 것이 힘들었다. 그리고 논리가 안 맞으면 자꾸 isEmpty나 pop할 때 에러가 난다.