분류
스택／큐
문제
일반적인 프린터는 인쇄 요청이 들어온 순서대로 인쇄합니다. 그렇기 때문에 중요한 문서가 나중에 인쇄될 수 있습니다. 이런 문제를 보완하기 위해 중요도가 높은 문서를 먼저 인쇄하는 프린터를 개발했습니다. 이 새롭게 개발한 프린터는 아래와 같은 방식으로 인쇄 작업을 수행합니다.
- 인쇄 대기목록의 가장 앞에 있는 문서(J)를 대기목록에서 꺼냅니다.
- 나머지 인쇄 대기목록에서 J보다 중요도가 높은 문서가 한 개라도 존재하면 J를 대기목록의 가장 마지막에 넣습니다.
- 그렇지 않으면 J를 인쇄합니다.
예를 들어, 4개의 문서(A, B, C, D)가 순서대로 인쇄 대기목록에 있고 중요도가 2 1 3 2 라면 C D A B 순으로 인쇄하게 됩니다.
내가 인쇄를 요청한 문서가 몇 번째로 인쇄되는지 알고 싶습니다. 위의 예에서 C는 1번째로, A는 3번째로 인쇄됩니다.
현재 대기목록에 있는 문서의 중요도가 순서대로 담긴 배열 priorities와 내가 인쇄를 요청한 문서가 현재 대기목록의 어떤 위치에 있는지를 알려주는 location이 매개변수로 주어질 때, 내가 인쇄를 요청한 문서가 몇 번째로 인쇄되는지 return 하도록 solution 함수를 작성해주세요.
제한사항
현재 대기목록에는 1개 이상 100개 이하의 문서가 있습니다.
인쇄 작업의 중요도는 1~9로 표현하며 숫자가 클수록 중요하다는 뜻입니다.
location은 0 이상 (현재 대기목록에 있는 작업 수 - 1) 이하의 값을 가지며 대기목록의 가장 앞에 있으면 0, 두 번째에 있으면 1로 표현합니다.
입출력 예
| priorities | location | return |
|---|---|---|
| [2, 1, 3, 2] | 2 | 1 |
| [1, 1, 9, 1, 1, 1] | 0 | 5 |
아이디어
-
조건을 만족하면 빼고, 아니면 뒤로 옮겨야 하므로 deque의 pop과 rotate를 쓰려고 함
-
입출력 예의 두 번째 경우 같이, 같은 수가 여러 개일 때 문제에서 원하는 location을 체크해야 하므로 location 체크용 idx 덱을 따로 만들어줌
-
deque도
in이랑max()를 쓸 수 있는지 모르겠어서 vscode에서 아무 리스트로 테스트한 결과 문제 없이 프린트 되는 것 확인 -
현재 pop된 값이 남은 덱에서의 최댓값보다 크다면 idx에서도 해당 인덱스를
pop -
현재 pop된 값보다 큰 값이 있다면 현재 값을
append해줘서 뒤로 넣어주고, idx도rotate(-1)해줘서 뒤로 옮겨줌 -
배열 길이가 1만 들어올 때 예외 처리를 안 해줘서 한 번 틀렸음
내 코드
from collections import deque
def solution(priorities, location):
idx = deque([i for i in range(len(priorities))] ) # 인덱스 기억용 리스트
deq = deque(priorities)
answer = 0
m = max(deq)
while location in idx:
n = deq.popleft()
if len(deq) == 0: # priorities 길이가 1일 때 예외처리
answer += 1
break
if n >= m: # 제일 중요도 높으면
idx.popleft() # 인쇄
answer += 1
m = max(deq)
else:
deq.append(n)
idx.rotate(-1)
return answer-
테스트 케이스 2, 5, 18 틀리길래 질문하기에서 찾아보니까 priorities 길이 1일 때를 안 챙겨준 거였음
-
priorities = [1] 이 들어왔을 때, 한 번 popleft() 하고 나면 덱이 비기 때문에 max()에 빈 덱이 들어가서 오류가 나는 거!
-
처음에 if n >= max(deq)로 매번 max 계산되게 했는데 max(deq)를 m으로 선언하고, 중요도 큰 게 pop 될 때만 다시 max 값 갱신해서 체크하도록 변경했음 → 쪼금이라도 빨라졌겠지
다른 사람 코드1
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-
튜플로 넣는 거 Do it 책에서 백준 11003번 설명할 때 보여준 방식인데 까먹고 있었다! 이렇게 하면 idx 덱 하나 줄일 수 있음
-
any는 생각도 못 했다 나한테 any는 SQL 뿐..^,ㅠ.. -
그래도 매번 any로 순회하는 것보다는 max 값 갱신이 빠를 것 같다고 생각했는데 vscode로 시간 찍어보면 그건 또 아닌 것 같기도 하고...?
다른 사람 코드2
for _ in range(t):
n, loc = map(int, input().rstrip().split())
priorities = list(map(int, input().rstrip().split()))
queue = [(i, priorities[i]) for i in range(len(priorities))]
priorities.sort(reverse=True) # 중요도 리스트를 따로 정렬 -> max 할 필요 없게
cnt = 0
for p in priorities: # 최댓값부터 보게 됨
while queue[0][1] != p: # 첫 번째 값이 최댓값이 아니면
tmp = queue.pop(0) # 빼내고 나서
queue.append(tmp) # 뒤로 넣음
cnt += 1 # while 빠져 나왔다는 거 == 최댓값이 빠져나왔다는 거, pop은 안 함
if queue.pop(0)[0] == loc: # pop 해서 인덱스 보니까 원하는 인덱스면
break # 끝
print(cnt)-
백준에서 똑같은 문제 푼 사람 거
-
이거는 리스트로 한 거고 중요도 값마다 while문 도니까 더 느릴 거라고 생각했는데 얘도 딱히 그건 아닌 것 같다 ...?
테스트케이스에 큰 값을 안 넣어봐서 그런가 돌릴 때마다 서로 비슷한 시간으로 나와서 속도 차이를 잘 모르겠는데.. 내 코드가 구리다고 생각은 안 할라고 ....~
