summaryrefslogtreecommitdiff
path: root/7-1.hs
diff options
context:
space:
mode:
Diffstat (limited to '7-1.hs')
-rw-r--r--7-1.hs22
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