22 lines
486 B
Haskell
22 lines
486 B
Haskell
{-# LANGUAGE OverloadedStrings #-}
|
|
|
|
import Data.List (nub)
|
|
import Lib (readFile')
|
|
import Data.Text (Text)
|
|
import qualified Data.Text as T
|
|
|
|
main :: IO ()
|
|
main = do
|
|
input <- readFile' "day6.in"
|
|
putStr "Q1: "
|
|
print $ parse 4 input
|
|
putStr "Q2: "
|
|
print $ parse 14 input
|
|
|
|
group :: Int -> Text -> [Text]
|
|
group _ "" = []
|
|
group n xs = T.take n xs : group n (T.tail xs)
|
|
|
|
parse :: Int -> Text -> Int
|
|
parse n = (+ n) . length . takeWhile ((< n) . length) . map (nub . T.unpack) . group n
|