X-Git-Url: https://git.njae.me.uk/?a=blobdiff_plain;f=advent14c.hs;h=af8a4afefe69ebbe5b9978e75c974127cc0d97ed;hb=e78bf6ce47d6ce753313cfe6a7d669e10b716d66;hp=8f1eb6b506c58dbbd280f270b410a6a9850bfb3b;hpb=3fef2ec4e8e9981203b1df48f986999217774fc0;p=advent-of-code-16.git

diff --git a/advent14c.hs b/advent14c.hs
index 8f1eb6b..af8a4af 100644
--- 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,22 +27,19 @@ 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
 
 substrings :: Int -> [a] -> [[a]]
 substrings l = filter (\s -> (length s) == l) . map (take l) . tails
-