feat: Year 2013 mergeHeaps

MMZK1526 · Jan 12, 2025 · e55f680 · e55f680
1 parent dcdd9d2
commit e55f680
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/src/Year2013/Exam.hs b/src/Year2013/Exam.hs
@@ -31,8 +31,13 @@ extractMin :: Ord a => BinHeap a -> a
 extractMin = minimum . map key
 
 mergeHeaps :: Ord a => BinHeap a -> BinHeap a -> BinHeap a
-mergeHeaps 
-  = undefined
+mergeHeaps h1@(t1 : ts1) h2@(t2 : ts2)
+  | r1 < r2   = t1 : mergeHeaps ts1 h2
+  | r1 > r2   = t2 : mergeHeaps h1 ts2
+  | otherwise = mergeHeaps [combineTrees t1 t2] (mergeHeaps ts1 ts2)
+  where
+    (r1, r2) = (rank t1, rank t2)
+mergeHeaps h1 h2 = h1 <> h2 -- > To reach here, one of the heaps must be empty
 
 insert :: Ord a => a -> BinHeap a -> BinHeap a
 insert