7-1

1 files changed, 22 insertions, 0 deletions

diff --git a/7-1.hs b/7-1.hs
new file mode 100644
index 0000000..81a6ba5
--- /dev/null
+++ b/7-1.hs
@@ -0,0 +1,22 @@
+combine :: [Int] -> [Int]
+combine (x:xs) = foldl f [x] xs
+  where
+    f :: [Int] -> Int -> [Int]
+    f ns n = [(+), (*)] <*> ns <*> [n]
+
+check :: Int -> [Int] -> Bool
+check x xs = elem x (combine xs)
+
+parseLine :: String -> (Int,[Int])
+parseLine s = (read $ init x, map read xs)
+  where
+    (x:xs) = words s
+
+result :: [(Int,[Int])] -> Int
+result xs = sum . map fst . filter snd
+  $ zip (map fst xs) (map (uncurry check) xs)
+
+main :: IO ()
+main = do
+  ls <- lines <$> getContents
+  print . result . map parseLine $ ls