Remove use of head.

chungyc · Jun 1, 2024 · a05b6b3 · a05b6b3
1 parent 536779b
commit a05b6b3
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/test/Problems/P98Spec.hs b/test/Problems/P98Spec.hs
@@ -1,5 +1,3 @@
-{-# OPTIONS_GHC -Wno-x-partial -Wno-unrecognised-warning-flags #-}
-
 {-|
 Copyright: Copyright (C) 2023 Yoo Chung
 License: GPL-3.0-or-later
@@ -55,7 +53,9 @@ spec = parallel $ do
 
 getRows :: [[Bool]] -> [[Int]]
 getRows = map lengths
-  where lengths bs = map length $ filter head $ group bs
+  where lengths bs = map length $ filter occupied $ group bs
+        occupied (v:_) = v
+        occupied [] = False
 
 getColumns :: [[Bool]] -> [[Int]]
 getColumns = getRows . transpose