Keyboards can be rectangular, not necessarily square

[advent-of-code-16.git] / advent02.hs

diff --git a/advent02.hs b/advent02.hs
index 99df10c23390948b8ff3adaf72c3493a3796f041..3ea1559d4646e4dc5cb74da01fa2d638d97ad6c2 100644 (file)
--- a/advent02.hs
+++ b/advent02.hs
@@ -21,7 +21,7 @@ kb2 = ["xxxxxxx",
 enumerate = zip [0..]
 
 mkKeyboard :: [String] -> Keyboard
-mkKeyboard kb = array ((0, 0), (length kb - 1, length kb - 1))
+mkKeyboard kb = array ((0, 0), (length kb - 1, length (kb!!0) - 1))
     [((i, j), c) | (i, r) <- enumerate kb, (j, c) <- enumerate r]
 
 keyboard1 = mkKeyboard kb1