Tidying

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

diff --git a/advent14c.hs b/advent14c.hs
index 8f1eb6b506c58dbbd280f270b410a6a9850bfb3b..a43e37fe4233d0a13ae4c9446fcc0cdaf966ed27 100644 (file)
--- a/advent14c.hs
+++ b/advent14c.hs
@@ -1,6 +1,4 @@
-import Data.List (nub, tails, null)
-import Data.Hash.MD5 (md5s, Str(..))
-
+import Data.List (nub, tails)
 import Data.ByteString.Char8 (pack)
 import Crypto.Hash (hash, Digest, MD5)
 
@@ -12,16 +10,14 @@ main = do
         part1 
         part2
 
-
 part1 :: IO ()
-part1 = print $ head $ drop 63 $ filter (\i -> possibleStart sq i && confirmStart sq i) [0..]
+part1 = print $ head $ drop 63 $ filter (\i -> possibleKey sq i && confirmKey sq i) [0..]
     where sq = md5sequence
 
 part2 :: IO ()
-part2 = print $ head $ drop 63 $ filter (\i -> possibleStart sq i && confirmStart sq i) [0..]
+part2 = print $ head $ drop 63 $ filter (\i -> possibleKey sq i && confirmKey sq i) [0..]
     where sq = md5sequenceS
 
-
 getHash :: String -> String
 getHash bs = show (hash $ pack bs :: Digest MD5)
 
@@ -31,19 +27,17 @@ md5sequence = [makeMd5 i | i <- [0..]]
 
 md5sequenceS :: [String]
 md5sequenceS = [makeMd5 i | i <- [0..]]
-    where makeMd5 i = stretch $ md5s (Str (salt ++ show i)) 
+    where makeMd5 i = stretch $ getHash (salt ++ show i)
           stretch h0 = foldr (\_ h -> getHash h) h0 [1..2016]
 
-possibleStart :: [String] -> Int-> Bool
--- possibleStart s i = not $ null $ repeats 3 $ s!!i
-possibleStart s = not . null . repeats 3 . ((!!) s)
+possibleKey :: [String] -> Int-> Bool
+possibleKey s = not . null . repeats 3 . ((!!) s)
 
-confirmStart :: [String] -> Int -> Bool
-confirmStart s i = any (confirmation) $ take 1000 $ drop (i+1) s
+confirmKey :: [String] -> Int -> Bool
+confirmKey s i = any (confirmation) $ take 1000 $ drop (i+1) s
     where c = head $ repeats 3 $ s!!i
           confirmation m = c `elem` (repeats 5 m)
                               
-
 repeats :: Int -> String -> [String]
 repeats n = filter (null . tail) . map (nub) . substrings n