feat(aoc-core): add PriorityQueue (#32)

leohidalgo · Jan 15, 2025 · d0260f8 · d0260f8
1 parent 28ec761
commit d0260f8
Showing 1 changed file with 22 additions and 0 deletions.
diff --git a/Sources/AOCCore/PriorityQueue.swift b/Sources/AOCCore/PriorityQueue.swift
@@ -0,0 +1,22 @@
+import Foundation
+
+public struct PriorityQueue<Element, Cost: Comparable> {
+
+    public var isEmpty: Bool { queue.isEmpty }
+
+    private var queue: [(element: Element, cost: Cost)] = []
+
+    public init() { }
+
+    mutating
+    public func append(_ element: Element, _ cost: Cost) {
+        queue.append((element, cost))
+
+        queue.sort { $0.cost < $1.cost }
+    }
+
+    mutating
+    public func removeFirst() -> (element: Element, cost: Cost) {
+        queue.removeFirst()
+    }
+}