aboutsummaryrefslogtreecommitdiff
path: root/Puzzle1Better.hs
diff options
context:
space:
mode:
authorYuchen Pei <me@ypei.me>2017-12-01 12:05:49 +0100
committerYuchen Pei <me@ypei.me>2017-12-01 12:05:49 +0100
commite1e59998c9b18ad262ddc3efbfa1c5bb972b86f1 (patch)
tree3ecf42c3153ad7079d709d79df03e2e1a5188e21 /Puzzle1Better.hs
first commit
Diffstat (limited to 'Puzzle1Better.hs')
-rw-r--r--Puzzle1Better.hs5
1 files changed, 5 insertions, 0 deletions
diff --git a/Puzzle1Better.hs b/Puzzle1Better.hs
new file mode 100644
index 0000000..d8d0155
--- /dev/null
+++ b/Puzzle1Better.hs
@@ -0,0 +1,5 @@
+import Data.Char (digitToInt)
+
+solve n xs = sum $ zipWith (\a b -> if a == b then a else 0) xs (drop n $ cycle xs)
+solve1 xs = solve 1 (digitToInt <$> xs)
+solve2 xs = solve ((length xs) `div` 2) (digitToInt <$> xs)