BOJ: #1966 - 프린터 큐

14:01

처음에 문서를 저장할 때, 문서의 중요도와 초기 인덱스를 같이 저장한다. 그런 다음, 큐가 빌 때까지 큐에서 가장 중요도가 높은 값을 구하고 첫번째 문서의 중요도와 비교한다. 같으면 answer 배열에 삽입하고, 같지 않으면 큐의 맨 뒤로 보낸다.
큐가 비면, answer 배열에서 초기 인덱스를 갖고 있는 문서의 위치를 찾아서 + 1 해서 출력한다.

문제

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

내가 작성한 코드

Swift

let t = Int(readLine()!)!
for _ in 0..<t {
    let n = readLine()!.split(separator: " ").map { Int(String($0))! }
    var importance = readLine()!.split(separator: " ").enumerated().map { (index, element) in
        return [Int(String(element))!, index]}
    var answer = [[Int]]()
    while !importance.isEmpty {
        if let first = importance.first, let max = importance.max(by: { $0[0] < $1[0] }) {
            if first[0] == max[0] {
                answer.append(importance.removeFirst())
            } else {
                importance.append(importance.removeFirst())
            }
        }
    }
 
    if let find = answer.firstIndex(where: { $0[1] == n[1] }) {
        print(find + 1)
    }
}