[Swift] 백준 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 이하의 정수이고, 중요도가 같은 문서가 여러 개 있을 수도 있다.

출력

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

접근

Swift에는 removeFirst() 나 removeLast() 같은 메소드를 만들어놔서 Queue를 따로 구현하지 않아도 충분히 문제를 풀 수 있다.
처음에 문제 대충 읽었을 때는 이거 우선순위 큐 문제인가? 했는데 다행히 그건 아니었다. 그냥 조건에 따라 분기를 나눠주고, while문을 돌면서 내가 찾는 인덱스가 몇번째로 인쇄되는지 확인하면 된다.
이를 구현하기 위해 각 인덱스의 우선순위를 입력받은 prior 배열을 오름차순으로 정렬했다.
그 다음 queue 가 텅 빌때까지 while문을 돌리면서
만약 queue 의 가장 첫 문서의 중요도가 가장 높지 않다면(if) queue 의 가장 뒤로 빼주고, 가장 높다면(else) 내가 찾고 있는 인덱스인지 확인한다.
중요도가 가장 높은데 만약 내가 찾고 있는 인덱스라면 break 해주고, count 를 출력한다. 아니라면 그냥 queue 에서 pop 해주고, prior 배열에서도 제거한다. 이후 count 를 1 증가시킨다.

코드

//
//  main.swift
//  ProblemSolving_1966
//
//  Created by YOUJIM on 2023/06/30.
//

import Foundation

for _ in 0..<Int(readLine()!)! {
    let nm = readLine()!.split(separator: " ").map { Int($0)! }
    var queue = [[Int]]()
    var prior = readLine()!.split(separator: " ").map { Int($0)! }
    var count = 1
    
    for i in 0..<nm[0] { queue.append([i]) }
    for i in 0..<prior.count { queue[i].append(prior[i]) }
    
    prior.sort()
    
    while !queue.isEmpty {
        if queue[0][1] != prior.last {
            queue.append(queue.removeFirst())
        }
        else {
            if queue[0][0] == nm[1] { break }
            queue.removeFirst()
            prior.removeLast()
            count += 1
        }
    }
    
    print(count)
    
}

배운 점

• 헷갈리면 주석을 적어가면서 하기! 특히 분기를 나눠야 하는 부분에서는 주석 적어놓는 게 도움이 된다. 물론 나중에는 이런 거 없이 해야겠지만 지금은 연습하는 단계니까 적어가면서 하는 게 좋은 것 같다.
• 항상 느끼지만 변수명 잘 설정해놓는 게 생각보다 정말정말 중요하다. 처음 입력받을 때 input 이런 식으로 변수를 설정했더니 나중에 코드를 보니까 저게 뭔지 하나도 감이 안왔다. 난 특히 한줄을 주르륵 입력받고 split() 으로 나눠서 쓰니까 더 헷갈리는 것 같은데... 결국 나중에 변수명을 전부 리팩토링해줬다.

---

요즘 백준 풀때 집중을 잘 못한다. 이 문제도 그렇게 오래 걸릴 문제는 아니었는데 한 50분 걸려서 풀었다.
백준을 잠깐 쉴까 했는데 그러기엔 그동안 해온 게 아깝고 ... 며칠은 브론즈 문제 풀면서 날먹도 좀 했다. 사실 이럴 때는 쉬면 그대로 풀어져서 꾸준히 푸는 게 낫긴 한데 ㅜㅜ 적당히 난이도 조절해가면서 이어가보겠다!