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

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

 

---

 

풀이 과정

1. plans를 시작 시간을 기준으로 오름차순 정렬해준다.
2. 시간 계산을 편하게 하기 위해 시작 시간의 hh:mm을 분 단위로 계산해 교체한다.
3. plans를 순회하며 차례로 종료 시간과 다음 시작 시간을 비교한다.
   • 종료 시간이 다음 시작 시간보다 클 경우, 기다려야 하므로 wait에 추가해준다.
   • 종료 시간이 다음 시작 시간보다 작거나 같을 경우, answer에 추가해준다.
     • 여유 시간을 구하고, wait에 중단된 과제가 있고 여유 시간이 1 이상일 때 가장 최근에 중단된 과제를 재개한다.
4. 순회가 끝나고 wait에 중단된 과제가 남아 있다면, 끝에서부터 차례로 answer에 추가해준다.

 

 

1. 일부분 오답

def solution(plans):
    answer = []
    wait = []

    plans.sort(key=lambda x: x[1])

    for p in plans:
        sub, start_time, play = p
        # hh:mm => 분 단위로 바꿔주기
        hour, minute = map(int, start_time.split(":"))
        cal_time = hour * 60 + minute
        p[1] = cal_time
        p[2] = int(p[2])

    for idx in range(len(plans)):
        # 종료 시간 구하기
        total = plans[idx][1] + plans[idx][2]
        if idx + 1 < len(plans):
            # 다음 시작 시간
            next_value = plans[idx + 1][1]
            # 종료 시간 > 다음 시작 시간 => 기다려야 함
            if total > next_value:
                # wait에 추가
                wait.append([idx, plans[idx][0]])
                # play_time 줄여주기
                plans[idx][2] -= next_value - plans[idx][1]
            else:
                # 끝났으므로 answer에 추가
                answer.append(plans[idx][0])
                # 다음 시작 시간까지 남은 여유시간
                left = next_value - total
                # 기다리고 있는 과제가 있고, 다음 시작 시간까지 1분 이상 남았다면
                if wait and left > 0:
                    # 가장 최근 대기한 과제
                    item = wait[-1]
                    # 줄일 수 있는 시간 계산
                    reduce_playtime = plans[item[0]][2] - left
                    # 여유 시간 내에 과제를 모두 끝내지 못했다면 남은 시간 갱신
                    if reduce_playtime > 0:
                        plans[item[0]][2] = reduce_playtime
                    else:
                        # 다 끝냈으면 answer에 추가
                        wait.pop()
                        answer.append(item[1])
        else:
            answer.append(plans[idx][0])

    # 남은 대기 중인 과제들 => 뒤에서부터 answer에 넣어주기(가장 최근에 들어옴 => 먼저 나감)
    for item in wait[::-1]:
        answer.append(item[1])

    return answer

- 4, 5, 6, 8, 9, 10번 테스트에서 실패가 떴다.

- 원인을 알 수 없어서 꽤 오래 고민했는데, 여유시간 내에 처리할 수 있는 중단된 과제가 1개 이상이라는 점을 고려하지 못했기 때문이었다.

- 따라서 아래와 같이 여유시간이 생긴다면, while문을 통해 최대한 중단된 과제를 해결하는 방식으로 코드를 수정했다.

 

2. 런타임 에러

def solution(plans):
    answer = []
    wait = []

    plans.sort(key=lambda x: x[1])

    for p in plans:
        sub, start_time, play = p
        # hh:mm => 분 단위로 바꿔주기
        hour, minute = map(int, start_time.split(":"))
        cal_time = hour * 60 + minute
        p[1] = cal_time
        p[2] = int(p[2])

    for idx in range(len(plans)):
        # 종료 시간 구하기
        total = plans[idx][1] + plans[idx][2]
        if idx == len(plans)-1:
            answer.append(plans[idx][0])
        else:
            # 다음 시작 시간
            next_value = plans[idx + 1][1]
            # 종료 시간 > 다음 시작 시간 => 기다려야 함
            if total > next_value:
                plans[idx][2] -= next_value - plans[idx][1]
                # wait에 추가
                wait.append([plans[idx][0], plans[idx][2]])
                # play_time 줄여주기
            else:
                # 끝났으므로 answer에 추가
                answer.append(plans[idx][0])
                # 다음 시작 시간까지 남은 여유시간
                left = next_value - total
                # 기다리고 있는 과제가 있고, 다음 시작 시간까지 1분 이상 남았다면
                if wait and left > 0:
                    for item in wait[::-1]:
                        title, playtime = item
                        if playtime <= left:
                            wa
                    while left > 0:
                        # 가장 최근 대기한 과제
                        title, playtime = wait[-1]
                        # 줄일 수 있는 시간 계산
                        # 가장 최근 중단한 과제가 여유 시간 내에 끝낼 수 있다면 => answer에 추가
                        if playtime <= left:
                            wait = wait[:-1]
                            answer.append(title)
                            left -= playtime
                        # 가장 최근 중단한 과제가 여유 시간 내에 다 끝낼 수 없다면 => 남은 시간 갱신
                        else:
                            wait[-1][1] -= left
                            break

    # 남은 대기 중인 과제들 => 뒤에서부터 answer에 넣어주기(가장 최근에 들어옴 => 먼저 나감)
    for item in wait[::-1]:
        answer.append(item[0])

    return answer

 

- 하지만 이번엔 런타임에러가 났다 🥲

- while문을 for문으로 바꾸어 실행 속도를 줄여주었다.

 

3. 최종 정답

def solution(plans):
    answer = []
    wait = []

    plans.sort(key=lambda x: x[1])

    for p in plans:
        sub, start_time, play = p
        # hh:mm => 분 단위로 바꿔주기
        hour, minute = map(int, start_time.split(":"))
        cal_time = hour * 60 + minute
        p[1] = cal_time
        p[2] = int(p[2])

    for idx in range(len(plans)):
        # 종료 시간 구하기
        total = plans[idx][1] + plans[idx][2]
        if idx == len(plans)-1:
            answer.append(plans[idx][0])
        else:
            # 다음 시작 시간
            next_value = plans[idx + 1][1]
            # 종료 시간 > 다음 시작 시간 => 기다려야 함
            if total > next_value:
                plans[idx][2] -= next_value - plans[idx][1]
                # wait에 추가
                wait.append([plans[idx][0], plans[idx][2]])
                # play_time 줄여주기
            else:
                # 끝났으므로 answer에 추가
                answer.append(plans[idx][0])
                # 다음 시작 시간까지 남은 여유시간
                left = next_value - total
                # 기다리고 있는 과제가 있고, 다음 시작 시간까지 1분 이상 남았다면
                if wait and left > 0:
                    temp = wait[:]
                    # 여유 시간 내에 해결할 수 있는 과제가 1개 이상일 수도 있음!!
                    for idx in range(len(temp)-1, -1, -1):
                        title, playtime = temp[idx]
                        if playtime < left:
                            wait.pop(idx)
                            answer.append(title)
                            left -= playtime
                        elif playtime == left:
                            wait.pop(idx)
                            answer.append(title)
                            break
                        else:
                            wait[idx][1] -= left
                            break

    # 남은 대기 중인 과제들 => 뒤에서부터 answer에 넣어주기(가장 최근에 들어옴 => 먼저 나감)
    for item in wait[::-1]:
        answer.append(item[0])

    return answer

- 통과!