diff options
author | Tom Smeding <tom@tomsmeding.com> | 2021-12-03 20:28:40 +0100 |
---|---|---|
committer | Tom Smeding <tom@tomsmeding.com> | 2021-12-03 20:28:40 +0100 |
commit | e0632dd35e0dd6491397514d09ad273f4d1e0392 (patch) | |
tree | 55fd6dd3f8e693290d8aed440820bc37f9f0fbbc /2021/3.hs | |
parent | f2f2cea14262c702a25325c8e6312687331be419 (diff) |
3
Diffstat (limited to '2021/3.hs')
-rw-r--r-- | 2021/3.hs | 25 |
1 files changed, 25 insertions, 0 deletions
diff --git a/2021/3.hs b/2021/3.hs new file mode 100644 index 0000000..98ca15c --- /dev/null +++ b/2021/3.hs @@ -0,0 +1,25 @@ +module Main where + +import Data.Bits +import Data.List + +import Input + + +readBin :: [Bool] -> Int +readBin = sum . zipWith (*) (iterate (*2) 1) . reverse . map fromEnum + +main :: IO () +main = do + inp <- map (map (== '1')) <$> getInput 3 + let n = length (head inp) + let majority s = 2 * length (filter id s) > length s + let gamma = readBin $ map majority $ transpose inp + epsilon = complement gamma .&. (bit n - 1) + print (gamma * epsilon) + let filterstep _ _ [x] = [x] + filterstep target idx cands = filter ((== target) . (!! idx)) cands + perform targetrule = foldl' (\cs i -> filterstep (targetrule cs i) i cs) inp [0 .. n - 1] + oxygen = perform $ \cs i -> 2 * length (filter id (map (!! i) cs)) >= length cs + co2 = perform $ \cs i -> 2 * length (filter id (map (!! i) cs)) < length cs + print (readBin (head oxygen) * readBin (head co2)) |