[프로그래머스][python] 프린터 - stack/queue

문제

• 프로그래머스: 프린터
• Level 2
• Stack/queue
• 문제링크

문제 설명

일반적인 프린터는 인쇄 요청이 들어온 순서대로 인쇄합니다. 그렇기 때문에 중요한 문서가 나중에 인쇄될 수 있습니다. 이런 문제를 보완하기 위해 중요도가 높은 문서를 먼저 인쇄하는 프린터를 개발했습니다. 이 새롭게 개발한 프린터는 아래와 같은 방식으로 인쇄 작업을 수행합니다.

1. 인쇄 대기목록의 가장 앞에 있는 문서(J)를 대기목록에서 꺼냅니다.
2. 나머지 인쇄 대기목록에서 J보다 중요도가 높은 문서가 한 개라도 존재하면 J를 대기목록의 가장 마지막에 넣습니다.
3. 그렇지 않으면 J를 인쇄합니다.

현재 대기목록에 있는 문서의 중요도가 순서대로 담긴 배열 priorities와 내가 인쇄를 요청한 문서가 현재 대기목록의 어떤 위치에 있는지를 알려주는 location이 매개변수로 주어질 때, 내가 인쇄를 요청한 문서가 몇 번째로 인쇄되는지 return 하도록 solution 함수를 작성해주세요.

나의 코드

순환 stack을 생각해서 풀었다.
1. 첫번째 인덱스는 고정 시키고 나머지의 리스트에서 더 큰 숫자가 있으면 해당 인덱스를 리스트의 가장 뒤로 보내준다.
2. 순환을 계속하며 첫 번째 인덱스가 가장 큰 숫자일 시, 스택에서 pop 시켜준다. Count를 증가시켜준다.
3. 모든 순환이 발생할 시, 내가 가지고 있는 문서의 인덱스 번호도 업데이트 해준다.
4. Stack이 빌 때까지 반복하며, 내가 가지고 있는 문서의 인덱스 번호가 0일때 pop이 되었다면 탐색 완료.

def solution(priorities, location):
    cnt = 0
    start = 0
    current_location = location
    while len(priorities) > 0:
        if len(priorities) == 1:
            return cnt + 1
        
        if priorities[start] < max(priorities[start + 1: len(priorities) + 1]):
            priorities.append(priorities[start])
            priorities.pop(start)
            if current_location == 0:
                current_location = len(priorities) - 1
            else:
                current_location -= 1
        else:
            priorities.pop(start)
            cnt += 1
            if current_location == 0:
                break
            else:
                current_location -= 1
            
    return cnt

예외처리:

• 스택의 길이가 1이 되는 경우에는, 더 이상 뒤에 있는 리스트 슬라이스와 비교할 수 없다. 따라서 따로 조건 처리를 해주었다.