전공/백준
1966
yha97
2022. 9. 27. 14:50
날짜 : 2022. 09. 27
사용 언어 : python
문제
코드
import sys
test = int(sys.stdin.readline())
for _ in range(test):
n, m = map(int, sys.stdin.readline().split()) # 문서의 개수, 현재 queue 에서의 순서
q = []
nums = []
t = list(map(int, sys.stdin.readline().split()))
for i in range(len(t)):
q.append([i, t[i]])
nums.append(t[i])
# print(q)
# print(nums)
check = q[m]
nums.sort(reverse=True) # 중요도 내림차순 정렬
cnt = 0 # 출력순서
# print(nums)
while True:
if q[0][1] == nums[0]: # 가장 큰 중요도인 경우 출력
# print("%d 출력 가능"%(q[0][1]))
cnt += 1
if q[0] == check:
break
del q[0] # 출력 후 삭제
del nums[0] # 가장 큰 숫자 출력 완료
else: # 중요도가 가장 높지 않은 경우(가장 뒷편으로 이동)
# print("%d 뒤로 이동"%(q[0][1]))
t1 = q.pop(0) # 제일 앞 요소 삭제
# print(q)
q.append(t1)
print(cnt)
알게된 점
- 맨 처음 queue의 순서와 중요도를 함께 저장하는 list를 만들고 그 리스트를 저장하는 리스트를 만들었다.
- 그리고 while문을 통해 출력하면 pop, 아닌 경우 뒤로 넣는 방식을 통해 문제를 해결해 나갔다.
- 중간에 최대 중요도를 어떻게 처리할까 고민한 결과 중요도만을 고려하는 리스트를 따로 만든 후 내림차순으로 정렬, 출력할 때마다 del 함수를 통해 최댓값을 삭제해 나갔다
- pop() 메소드는 리스트의 가장 마지막 요소를 반환 및 삭제하는 메소드다.
(FIFO로 착각해서 첫 번째 요소를 뽑아내는 줄 알았다.)
- 알바하면서는 문제이해도 되게 안됐는데 막상 집에서 편하게 하니까 한번에 성공해서 좀 당황했다.
참고 사이트
-