[프로그래머스 / Python] 프로세스

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

프로그래머스

 

오 우선순위 큐 아니야 하면서 heapq로 최대힙 구현해서 풀려고 했는데

보니깐 아니였다...

 

그래서 그냥 큐를 만들어서 나온 로직대로 구현했다.

그런데 역시 python 답게 더 짧게 구하는 풀이가 있었다!

바로 any랑 enumerate 사용!

 

멋진 풀이다.

from collections import deque

def solution(priorities, location):
    h = [[priorities[i], i] for i in range(len(priorities))]
    q = deque(h)
    answer = 0
    
    while q :
        t = q.popleft()
        chk = True
        for i in q :
            if i[0] > t[0] :
                q.append(t) 
                chk = False
                break
        if chk : answer+=1
        if chk and t[1] == location :
            return answer

 

any 함수, enumerate 이용하기

def solution(priorities, location):
    queue =  [(i,p) for i,p in enumerate(priorities)]
    answer = 0
    while True:
        cur = queue.pop(0)
        if any(cur[1] < q[1] for q in queue):
            queue.append(cur)
        else:
            answer += 1
            if cur[0] == location:
                return answer