[프로그래머스 lv 2] 프로세스 (스택/큐)

아이디어 : deque를 사용하면 편할 것 같다.

주의할 점 : 프로세스가 '실행'된 값이 answer이다.(큐를 옮긴 횟수가 아님)

 

# 첫 번째 시도

from collections import deque

def solution(priorities, location):
    answer = 0
    queue = deque(priorities)
    while True:
        if location==0:
            if queue[location]==max(priorities):
                answer+=1
                return answer
            else:
                target = queue.popleft()
                queue.append(target)
                location = len(queue)-1
        else:
            if queue[0]==max(priorities):
                answer+=1
                queue.popleft()
                location -= 1
            else:
                target = queue.popleft()
                queue.append(target)
                location -= 1

 

여기서 while문을 계속 도는 것 같은데 어떤 부분이 문제가 있는지 잘 못 찾겠다. -> max(queue)가 아니라 max(priorities)를 찾으니... 당연히 종료가 안되는 거였다.

 

# 두 번째 시도

 

from collections import deque

def solution(priorities, location):
    answer = 0
    queue = deque(priorities)
    while True:
        if location==0:
            if queue[0]==max(queue):
                answer+=1
                return answer
            else:
                target = queue.popleft()
                queue.append(target)
                location = len(queue)-1
        else:
            if queue[0]==max(queue):
                answer+=1
                queue.popleft()
                location -= 1
            else:
                target = queue.popleft()
                queue.append(target)
                location -= 1