프로그래머스 | 과제 진행하기 | 파이썬 풀이

문제

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

내 답안

def solution(plans):
    # 시간 분 단위 숫자화 후 오름차순 정렬
    for i in plans:
        time = i[1]
        hour = int(time[:2])
        minute = int(time[3:])
        i[1] = 60*hour + minute
        i[2] = int(i[2])
    plans = sorted(plans, key = lambda x : x[1]) 


    stack = [] # 일시 중단된 과목들
    fin = [] # 수행 완료된 과목들
    i = 0
    time = plans[0][1] # 현재 시간

    # 새 작업들 시간 순서대로 수행하기
    while i < len(plans) :
        cur = plans[i]
        cur_start_time = cur[1]
        cur_end_time = cur[1] + cur[2]
        next_start_time = 60*24 if len(plans) == i+1 else plans[i+1][1]

        # 현재 과목 시작가능한 시간이 아니라면
        if time < cur_start_time :
            # 스택안의 과제가 존재할 때
            if stack :
                name, left_time = stack.pop()
                # 끝낼 수 있다면
                if time + left_time <= cur_start_time :
                    fin.append(name)
                    time = time + left_time

                # 끝낼 수 없다면
                else :
                    duration = cur_start_time - time
                    stack.append((name, left_time - duration)) # 가능한만큼만 하고 다시 넣기
                    time = cur_start_time

            # 스택안의 과제가 없다면 현재 과제 바로 시작
            else :
                time = cur_start_time

        # 다음 과목 시작전까지 현재 과목을 끝낼 수 있다면
        elif (cur_end_time <= next_start_time) : 
            fin.append(cur[0])
            i += 1
            time = cur_end_time 

        # 현재 과목을 중단해야 한다면
        else :
            left_time = cur[2] - (next_start_time - cur_start_time)
            stack.append((cur[0], left_time))
            time = next_start_time
            i += 1


    #남은 스택 비우기
    while stack:
        temp = stack.pop()
        fin.append(temp[0])

    return fin

해설

풀이 과정은 다음과 같다.

1. 문자열 타입의 과목별 시작 시간과 playtime 값을 정수로 형변환을 적용하고 분 단위로 바꿔서 저장한다.

2. 오름차순으로 배열을 정렬한다.

3. 배열안의 과목들을 차례대로 수행한다.

• 과목들의 수행 가능성을 판단하기 위해 현재 시간을 나타내는 변수 time을 준비한다.
• 수행된 과목의 개수를 카운트할 변수 i를 준비한다.
• 현재 과목을 시작할 수 없다면, 즉 시간이 아직 안된 경우 아래의 케이스로 나뉜다.
  • 중단된 과제가 존재하고
    • 그 과제를 현재 과목을 시작하기 전까지 완료 가능
    • 완료 불가능
  • 중단된 과제가 존재하지 않음
    • 현재 과목 바로 시작하기
• 현재 과목을 시작할 수 있는 시간이 되었을 때, 아래의 두 케이스로 나뉜다.
  • 다음 과목 시작 전까지 현재 과목 완료 가능
  • 완료 불가능

4. 중단된 과목들이 남아있다면 차례대로(pop()) 수행한다.

피드백

아이디어는 빨리 떠올랐는데, 중간중간에 뻘짓;;을 많이해서 시간이 많이 걸렸다.
if 문을 많이 써야 할 때는 조건을 꼼꼼하게 생각해야한다. 어설프게 짜면 여기저기서 빵꾸나서 통과가 안됨..