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

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

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr

성능 요약

메모리: 79.8 MB, 시간: 6.16 ms

문제 설명

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

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

과제 계획을 담은 이차원 문자열 배열 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"] 순서로 과제를 마칩니다.

문제 풀이

1. 과제 리스트 생성 - 과제 시작 시간을 분으로 변환하고 시작시간순으로 오름차순 정렬

2. 과제 관리를 위한 스택 생성 - 스택에 진행 중인 과제 리스트들을 담고 새 작업을 순회하면서 현재 과제를 종료할지, 새 과제를 진행할지 비교

3. 과제 리스트를 순회하면서 현재 작업 중인 과제와 새 과제를 비교

스택이 비어있는 경우
- 새 과제 진행을 위해 스택에 push
스택이 비어있지 않은 경우
- 현재 시간 (stack에서 peek 한 값의 시작 시간 + 과제 진행 시간)이 새 과제 시간보다 같거나 빠른 경우
  - 현재 과제 종료 및 남아 있는 작업이 있다면 recursive 하게 pop
- 현재 시간이 새 과제 시간보다 늦은 경우
  - 새 작업 진행을 위해 스택에 새 작업을 push, 현재 과제에서 진행한 시간만큼 빼준다.

자바 코드

public String[] solution(String[][] plans) {
    Assignment[] arr = new Assignment[plans.length];
    for (int i = 0; i < plans.length; i++) {
        Assignment ass = new Assignment(plans[i][0], plans[i][1], plans[i][2]);
        arr[i] = ass;
    }

    Arrays.sort(arr, (o1, o2) -> {
        return o1.start - o2.start;
    });


    Stack<Assignment> stack = new Stack<>();  // 진행 중인 과제
    List<String> ans = new ArrayList<>();

    int curTime = -1;                         // 현재 시간 초기화

    for (int i = 0; i < arr.length; i++) {
        /* 진행 중인 과제가 없는 경우 */
        if (stack.isEmpty()) {
            stack.push(arr[i]);
            continue;
        }

        /* 진행 중인 과제와 새로운 과제가 있는 경우 */
        Assignment curAss = stack.peek();   // 진행중 과제
        Assignment newAss = arr[i];         // 새로운 과제

        // 새로운 과제의 시작시간과 진행중 과제의 종료 시간 비교
        curTime = curAss.start;

        // 현재 과제 시작 시간 + 작업 시간이 새로운 과제 시작 시간 보다 같거나 작은 경우
        if (curTime + curAss.time <= newAss.start) {
            recursivePop(stack, newAss, curTime, ans);
        } else {
            // 현재 작업 중단하고 작업한 시간 갱신 및 새 작업 시작
            curAss.time -= newAss.start - curTime;
        }
        stack.push(newAss);
    }

    /* 새로운 과제가 없는 경우 */
    while (!stack.isEmpty()) {
        ans.add(stack.pop().name);
    }
    return ans.toArray(new String[0]);
}

public void recursivePop(Stack<Assignment> stack, Assignment newAss, int curTime, List<String> ans) {
    if (stack.isEmpty()) {
        return;
    }
    Assignment curAss = stack.peek();   // 진행중 과제
    if (curTime + curAss.time <= newAss.start) {
        ans.add(stack.pop().name);
        recursivePop(stack, newAss, curTime + curAss.time, ans);
    } else {
        curAss.time -= newAss.start - curTime;
    }
}



static class Assignment {
    private String name;
    private int start;
    private int time;

    public Assignment(String name, String start, String time) {
        this.name = name;
        this.start = timeToMinute(start);
        this.time = Integer.parseInt(time);
    }

    public int timeToMinute(String start) {
        String[] arr = start.split(":");
        int h = Integer.parseInt(arr[0]) * 60;
        int m = Integer.parseInt(arr[1]);
        return h + m;
    }
}

자바스크립트 코드

let Assignment = class {
    constructor(name, start, time) {
        this.name = name;
        this.start = timeToMinute(start);
        this.time = parseInt(time);
    }

};

function timeToMinute(start) {
    const time = start.split(':');
    return parseInt(time[0]) * 60 + parseInt(time[1]);
};


function solution(plans) {
    var answer = [];
    let arr = [];

    for (let [name, start, time] of plans) {
        let ass = new Assignment(name, start, time); 
        arr.push(ass);
    }
    arr.sort((o1, o2)=> {
        return o1.start - o2.start;
    })
    let stack = [];
    let curTime = -1;

    for (let i=0; i < arr.length; i++) {
        if (stack.length == 0) {
            stack.push(arr[i]);
            continue;
        }

        let curAss = stack[stack.length - 1];
        let newAss = arr[i];

        curTime = curAss.start;

        if (curTime + curAss.time <= newAss.start) {
            recursivePop(stack, newAss, curTime, answer);
        } else {
            curAss.time -= newAss.start - curTime;
        }
        stack.push(newAss);
    }

    while (stack.length > 0) {
        const item = stack.pop();
        answer.push(item.name);
    }
    return answer;
};

function recursivePop(stack, newAss, curTime, answer) {
    if (stack.length == 0) {
        return;
    }
    let curAss = stack[stack.length - 1];
    if (curTime + curAss.time <= newAss.start) {
        answer.push(stack.pop().name);
        recursivePop(stack, newAss, curTime + curAss.time, answer);
    } else {
        curAss.time -= newAss.start - curTime;
    }
};

저작자표시