[알고리즘] 백준 1966 (파이썬 풀이)

    728x90

    https://www.acmicpc.net/problem/1966

     

    1966번: 프린터 큐

    여러분도 알다시피 여러분의 프린터 기기는 여러분이 인쇄하고자 하는 문서를 인쇄 명령을 받은 ‘순서대로’, 즉 먼저 요청된 것을 먼저 인쇄한다. 여러 개의 문서가 쌓인다면 Queue 자료구조에

    www.acmicpc.net

    문제

    여러분도 알다시피 여러분의 프린터 기기는 여러분이 인쇄하고자 하는 문서를 인쇄 명령을 받은 ‘순서대로’, 즉 먼저 요청된 것을 먼저 인쇄한다. 여러 개의 문서가 쌓인다면 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 이하의 정수이고, 중요도가 같은 문서가 여러 개 있을 수도 있다.

    출력

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

     

    해결코드

    from collections import deque
    
    t = int(input())
    for i in range(t):
          cnt = 0
          n,m = map(int, input().split())
          li = deque(map(int, input().split()))
          while(li):
                li_max = max(li)
                m -= 1
                pop = li.popleft()
                if(li_max != pop):
                      li.append(pop)
                      if(m<0):
                            m = len(li)-1
                else:
                      cnt += 1
                      if(m == -1):
                            print(cnt)
                            break

    문제풀이

    이 문제는 프린터의 인쇄 순서(우선순위)에 따라 먼저 큐에서 빠져나오는 원리를 이용한다.

    원하는 문서가 몇 번째로 빠져나오는지 출력하는 문제이다.

    먼저 deque을 이용해 계산시간을 최대한 줄여주도록 한다.

    int(input())을 통해 t에 테스트 케이스 수를 받아준다.

    테스트 케이스 수만큼 for문을 돌려준다.

    cnt(count)는 0으로 초기화 해주고, n,m을 map함수를 이용해 띄어쓰기로 받아준다.

    li라는 덱을 생성하여 그 안에 원소 값들을 받아준다.(프린터 안의 문서)

    while문을 돌리는데 li의 원소가 모두 없어질때까지 반복한다.

    while문 원리는 li안 가장 큰 값인 원소를

    li_max에 넣어놓고 m을 1씩 줄인다.

    그리고 리스트 맨 앞의 값을 제거하고 pop에 저자한다.

    pop과 li_max의 값이 같지 않다면 (방금 뺀 값이 최대값과 일치하지 않는다면)

    방금 뺀 값(pop)을 다시 li 뒤쪽에 추가한다.

    만약 m이 0보다 작다면(m값이 -1이라면) m값을 li 길이-1로 초기화(원하는 값이 안빠졌기 때문에)

    else문에서는 (pop과 li_max가 같은경우)

    우선순위가 가장 큰 것이 li에서 빠졌으므로 cnt값을 1올려준다.

    만약 m이 -1이 된다면 cnt값을 출력하고 while문을 탈출한다.

    728x90

    댓글