-import Data.List (sort)
-import Data.List.Split (splitOn)
-import Data.Array.IArray
+module Main(main) where
+import Data.Array.IArray
+-- Row 1 is top, column 1 is left
type Position = (Int, Int)
type Keyboard = Array Position Char
-kb1 = [['x', 'x', 'x', 'x', 'x'],
- ['x', '1', '2', '3', 'x'],
- ['x', '4', '5', '6', 'x'],
- ['x', '7', '8', '9', 'x'],
- ['x', 'x', 'x', 'x', 'x']]
-
+kb1 = ["xxxxx",
+ "x123x",
+ "x456x",
+ "x789x",
+ "xxxxx"]
kb2 = ["xxxxxxx",
"xxx1xxx",
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
keyboard2 = mkKeyboard kb2
findKey :: Keyboard -> Char-> Position
-findKey kb c = fst $ head $ filter (\(i, e) -> e == c) $ assocs kb
+findKey kb c = fst $ head $ filter (\a -> (snd a) == c) $ assocs kb
-- data Coord = One | Two | Three
-- deriving (Read, Show, Eq, Ord, Enum, Bounded)
-- -- minBound = Coord 1
-- -- maxBound = Coord 3
--- -- Row 1 is top, column 1 is left
-- data Position = Position Coord Coord
-- deriving (Show, Eq)
safeMove kb pos dir = maybeRevert kb pos (move pos dir)
move :: Position -> Char -> Position
-move (r, c) 'U' = (dec r, c)
-move (r, c) 'D' = (inc r, c)
-move (r, c) 'L' = (r, dec c)
-move (r, c) 'R' = (r, inc c)
+move (r, c) 'U' = (r-1, c)
+move (r, c) 'D' = (r+1, c)
+move (r, c) 'L' = (r, c-1)
+move (r, c) 'R' = (r, c+1)
maybeRevert :: Keyboard -> Position -> Position -> Position
maybeRevert kb oldPos newPos
| kb ! newPos == 'x' = oldPos
| otherwise = newPos
-
-numberOf p = '1'
--- numberOf :: Position -> Char
--- numberOf (Position One One) = '1'
--- numberOf (Position One Two) = '2'
--- numberOf (Position One Three) = '3'
--- numberOf (Position Two One) = '4'
--- numberOf (Position Two Two) = '5'
--- numberOf (Position Two Three) = '6'
--- numberOf (Position Three One) = '7'
--- numberOf (Position Three Two) = '8'
--- numberOf (Position Three Three) = '9'
-
--- | a `succ` that stops
-inc :: (Bounded a, Enum a, Eq a) => a -> a
-inc dir | dir == maxBound = maxBound
- | otherwise = succ dir
-
--- | a `pred` that stops
-dec :: (Bounded a, Enum a, Eq a) => a -> a
-dec dir | dir == minBound = minBound
- | otherwise = pred dir
\ No newline at end of file