[파이썬/Python] 백준 1966번 프린터 큐

[파이썬/Python] 백준 1966번 프린터 큐

1966번: 프린터 큐

---

문제

여러분도 알다시피 여러분의 프린터 기기는 여러분이 인쇄하고자 하는 문서를 인쇄 명령을 받은 ‘순서대로’, 즉 먼저 요청된 것을 먼저 인쇄한다. 여러 개의 문서가 쌓인다면 Queue 자료구조에 쌓여서 FIFO - First In First Out - 에 따라 인쇄가 되게 된다. 하지만 상근이는 새로운 프린터기 내부 소프트웨어를 개발하였는데, 이 프린터기는 다음과 같은 조건에 따라 인쇄를 하게 된다.

1. 현재 Queue의 가장 앞에 있는 문서의 ‘중요도’를 확인한다.
2. 나머지 문서들 중 현재 문서보다 중요도가 높은 문서가 하나라도 있다면, 이 문서를 인쇄하지 않고 Queue의 가장 뒤에 재배치 한다. 그렇지 않다면 바로 인쇄를 한다.

예를 들어 Queue에 4개의 문서(A B C D)가 있고, 중요도가 2 1 4 3 라면 C를 인쇄하고, 다음으로 D를 인쇄하고 A, B를 인쇄하게 된다.

여러분이 할 일은, 현재 Queue에 있는 문서의 수와 중요도가 주어졌을 때, 어떤 한 문서가 몇 번째로 인쇄되는지 알아내는 것이다. 예를 들어 위의 예에서 C문서는 1번째로, A문서는 3번째로 인쇄되게 된다.

 

입력

첫 줄에 테스트케이스의 수가 주어진다. 각 테스트케이스는 두 줄로 이루어져 있다.

테스트케이스의 첫 번째 줄에는 문서의 개수 N(1 ≤ N ≤ 100)과, 몇 번째로 인쇄되었는지 궁금한 문서가 현재 Queue에서 몇 번째에 놓여 있는지를 나타내는 정수 M(0 ≤ M < N)이 주어진다. 이때 맨 왼쪽은 0번째라고 하자. 두 번째 줄에는 N개 문서의 중요도가 차례대로 주어진다. 중요도는 1 이상 9 이하의 정수이고, 중요도가 같은 문서가 여러 개 있을 수도 있다.

출력

각 테스트 케이스에 대해 문서가 몇 번째로 인쇄되는지 출력한다.

예제 입력

3
1 0
5
4 2
1 2 3 4
6 0
1 1 9 1 1 1

예제 출력

1
2
5

풀이

import sys

# 테스트 케이스의 수
case = int(sys.stdin.readline())

for _ in range(case):
    # 문서의 개수 N,  몇 번째로 인쇄되었는지 궁금한 문서가 
    # 현재 Queue에서 몇 번째에 놓여 있는지를 나타내는 정수 M
    n, m = map(int, sys.stdin.readline().split())
    
    # N개 문서의 중요도
    priority = list(map(int, sys.stdin.readline().split()))
    
    # N개 문서의 기존 순서 저장
    index = [i for i in range(n)]
    
    # 몇 번째로 인쇄할지 카운트하는 변수
    count = 0
    
    while True:
        # 현재 문서가 중요도가 제일 높다면
        if priority[0] == max(priority):
            # 몇 번째로 출력 되는지 카운트
            count += 1
            # 궁금한 문서인지 확인
            if index[0] == m:
                print(count)
                break
            # 궁금한 문서가 아니라면 리스트에서 탈출
            else:
                del priority[0]
                del index[0]
        # 나머지 문서들 중 현재 문서보다 중요도가 높은 문서가 하나라도 있다면
        else:
            # 이 문서를 인쇄하지 않고 Queue의 가장 뒤에 재배치
            priority.append(priority[0])
            del priority[0]
            index.append(index[0])
            del index[0]

처음 자료구조에 대한 지식이 부족하여 del 메소드 대신 pop 메소드를 사용했었다.

 

그러나 상기 코드에서 del과 pop을 비교해보면, del은 인덱스를 이용하여 요소를 삭제하는 반면, pop은 인덱스를 이용하여 삭제하면서 삭제한 요소를 반환한다. 따라서 del 메서드는 삭제한 요소를 반환하지 않으므로 pop 메서드보다 불필요한 작업을 수행하지 않는다. 또한 del 메서드는 인덱스를 이용해 삭제하므로 평균적으로 pop 메서드보다 약간 더 빠르게 작동한다.

하지만 pop 메서드는 삭제한 요소를 반환하므로, 삭제한 요소를 사용해야 하는 경우에는 pop 메서드를 사용하는 것이 더 편리하다. 또한 pop 메서드는 인덱스를 지정하지 않으면 가장 마지막 요소를 삭제한다. 따라서 del 메서드와는 달리 인덱스를 지정하지 않아도 되는 경우에는 pop 메서드를 사용하는 것이 더 간단하다.

따라서, 이 코드에서는 삭제된 요소를 사용하지 않으므로 del 메서드를 사용하는 것이 더 효율적이다.

 

참고자료

파이썬 del 명령어 - list의 pop() 및 remove() 메소드 차이점