[Programmers] 과제 진행하기(Python)

문제

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

프로그래머스

---

풀이

한동안 코테 대비로 파이썬에만 집중해야겠다.

자바스크립트랑 자바는 나중에 다시 채워야지 ㅎㅎㅎ

 

Python

from collections import deque

def solution(plans):
    plans.sort(key=lambda x:x[1])
    answer, stack = [], []

    n, s, r = plans[0]
    p1, p2 = map(int, s.split(':'))
    s = p1*60 + p2
    
    stack.append((n, s, int(r)))
    
    for i in range(1, len(plans)):
        name, now, time = plans[i]
        n1, n2 = map(int, now.split(':'))
        start = n1 * 60 + n2
        
        while stack:
            pname, pstart, ptime = stack.pop()
            
            if pstart + ptime > start:
                stack.append((pname, start, pstart+ptime-start))
                break
            else:
                answer.append(pname)
                if stack:
                    pp1, pp2, pp3 = stack.pop()
                    stack.append((pp1, pstart+ptime, pp3))
                
        stack.append((name, start, int(time)))            
        
    while stack:
        a, b, c =stack.pop()
        answer.append(a)
        
    return answer