[Swift Algorithm] 이중우선순위큐 Programmers(Lv.3)

문제

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

코딩테스트 연습 - 이중우선순위큐

 

내 문제 풀이

func solution(_ operations:[String]) -> [Int] {
    var queue = [Int]()
    
    for i in operations {
        let command = i.split(separator: " ").map({String($0)})
        
        if command[0] == "I" { queue.append(Int(command[1])!) }
        else if command[0] == "D" && !queue.isEmpty {
            if command[1].contains("-") { queue.sort(by: >); queue.removeLast() }
            else { queue.sort(); queue.removeLast() }
        }
    }
    
    return queue.isEmpty ? [0, 0] : [queue.max()!, queue.min()!]
}

👉Heap 문제!

• queue 라는 배열을 생성하고 각 명령에 맞게 배열을 수정했음.
• I 명령어의 경우에는 바로 queue 배열의 마지막에 값을 삽입하였음.
• D 명령어의 경우에는 queue 배열이 비어있지는 않은지 확인 후 명령을 실행하게 함.
  숫자에 "-"가 붙어있는 경우 최솟값을 삭제해야하므로 queue 배열을 내림차순 정렬 후 removeLast 했음.
  그렇지 않은 경우에는 최댓값을 삭제해야하므로 queue 배열을 오름차순 정렬 후 removeLasrt 했음.
  (removeFirst를 하는 것을 피하기 위하여 위와 같이 정렬하였음.)

 

💡 피드백

• 레벨 3 치고는 굉장히 쉬운 문제였고, removeFirst만 피해주면 간단하게 풀이 가능했음.