프로그래머스 - 프린터(Swift)

프로그래머스 - 프린터(Swift)

 

 

 문제 설명

 

https://school.programmers.co.kr/learn/courses/30/lessons/42587

프로그래머스

 

 다른사람의 풀이

 

우선순위 큐 문제라는 것을 알았지만 어떻게 구현해야하는지, 이 문제에서는 어떻게 사용해야할지 몰라서 다른 사람의 코드를 참고했다. 

 

import Foundation

func solution(_ priorities:[Int], _ location:Int) -> Int {
    
    var queue: [(Int, Int)] = priorities.enumerated().map{ return ($0.offset, $0.element)}
    var priorityQueue = priorities.sorted(by: >)
    var result = 0
    
    while !queue.isEmpty {
        let first = queue.removeFirst()
        if first.1 == priorityQueue.first! {
            result += 1
            if first.0 == location {
                break
            }
            priorityQueue.removeFirst()  
        } else {
            queue.append(first)
        }
    }
    
    return result
}

 

문제의 location에 접근하기 위해서 enumerated한 후 map시켜서 배열의 형태로 만든 것을 큐로 정의했다. 여기서 enumerated만 사용하고 map은 사용하지 않는다면 EnumeratedSequence<Array<Int>>라는 새로운 타입이 있기 떄문에 map을 해줘야 한다.

 

두 번째로 우선순위 큐를 정의하는데 실제 우선순위 큐는 힙 자료구조를 이용해서 삽입 및 제거에 O(logN)을 따르는 자료구조이지만 결국 최대값을 맨 앞 인덱스에서 뽑는다는 맥락과 동일하게 sort로 간단하게 구현해주었다.

 

이제 이 두 개의 큐를 비교하면서 location과 queue의 offset이 동일해지는 시점의 result를 반환해주면 된다.