기록하고 까먹지 말기

1966 본문

전공/백준

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로 착각해서 첫 번째 요소를 뽑아내는 줄 알았다.)

- 알바하면서는 문제이해도 되게 안됐는데 막상 집에서 편하게 하니까 한번에 성공해서 좀 당황했다.

 

 

참고 사이트

 

'전공 > 백준' 카테고리의 다른 글

2960  (0) 2022.09.28
이코테 - 문자열 재정렬  (0) 2022.09.28
10773  (0) 2022.09.26
2864  (0) 2022.09.25
1541  (0) 2022.09.23