[프로그래머스] 프린터

안녕하세요. 밀쿄입니다.

오늘도 문제하나를 같이 풀어보겠습니다.

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

프로그래머스

이 문제는 우선순위 큐를 사용하는 문제입니다.

 

먼저 큐와 우선순위 큐 두 개를 만들어 줍니다.
큐는 배열의 인덱스와 함께 우선순위를 저장해줍니다,

그리고 우선순위큐는 우선순위를 정렬해서 저장해줍니다.

 

자 이렇게 해주고 나면 이제 당연하게

큐에서 하나 pop하고 우선순위큐에서 pop을 해서 우선순위를 비교해줍니다. 

우선순위가 같다면 당연히 인쇄를 해주고 리턴해줘야 할 값에 +1를 해주면 됩니다.

이 때 큐에 인덱스같이 내가 찾는 location 값이라면 리턴을 해주면 됩니다.

그리고 만약 같지 않다면 우선 순위가 더 높은 문서가 존재한다는 뜻이므로 큐에서 하나를 꺼내서 맨 끝에 push 해주시면 됩니다.

 

func solution(_ priorities:[Int], _ location:Int) -> Int {
    
    //인덱스랑 우선 순위가 같이 저장되는 큐
    var queue: [(Int, Int)] = []
    //우선순위큐( 우선순위 큐는 기본적으로 최대힙이므로 내림차순으로 정렬한다
    var priorityQueue = priorities.sorted(by : >)
    //결과적으로 리턴해줘야하는 값
    var result = 0
    
    //우선순위를 순회하면서 큐에 인덱스와 함께 우선순위를 저장한다.
    priorities.enumerated().forEach { (index, property) in
        queue.append((index, property))
    }
    
    //큐가 비어있을 때까지 순회를 한다.
    while !queue.isEmpty {
        
        // 큐의 맨 앞에 있는 우선순위와 우선순위큐의 첫번째 값을 비교한다.
        if queue.first!.1 == priorityQueue.first! {
            
            //큐의 맨앞에 있는 값의 인덱스가 내가 요청한 문서의 인덱스인지 확인한다.
            if queue.first!.0  == location {
                return result + 1
            }
            
            //우선순위의 값과 일치하므로 큐애 우선순위 큐에서 하나를 pop한다.
            result += 1
            queue.removeFirst()
            priorityQueue.removeFirst()
        } else {
            // 더 높은 우선순위의 값이 존재하므로 하나를 꺼내 맨 뒤에 push 한다.
            queue.append(queue.removeFirst())
        }
    }
    
    return result
}