프로그래머스 | 스택/큐 | 프로세스 (python)

https://school.programmers.co.kr/learn/courses/30/lessons/42587?language=python3

프로그래머스

 

코드

from collections import deque

def solution(priorities, location):
    queue = deque([(priority, idx) for idx, priority in enumerate(priorities)])
    cnt = 0
    
    while queue:
        curr_process = queue.popleft()
        
        if queue and curr_process[0] < max(queue, key = lambda x: x[0])[0]:
            queue.append(curr_process)
        else:
            cnt += 1
            if curr_process[1] == location:
                break
        
    
    return cnt

 

max값을 비교하는 과정에서 queue가 비어있는 상황에 런타임 에러 나서 queue 조건을 추가해줘야했다.

 

그래서

max값을 따로 저장하는 방식으로 바꾸는 게 좋을듯

from collections import deque

def solution(priorities, location):
    queue = deque([(priority, idx) for idx, priority in enumerate(priorities)])
    cnt = 0
    
    max_priority = max(priorities)
    
    while queue:
        curr_process = queue.popleft()
        
        if curr_process[0] < max_priority:
            queue.append(curr_process)
        else:
            cnt += 1
            if curr_process[1] == location:
                return cnt
            
            if queue:
                max_priority = max(queue, key=lambda x: x[0])[0]
    
    return cnt