blob: 40eaf9625cecec75823093e7d5288ad155405e5a (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
|
module Main where
import Data.Either
import Data.List
import Input
import IntCode
part1 :: [Integer] -> Integer
part1 program =
let settings = permutations [0..4]
outcome setting = foldr (.) id [\i -> head (snd (run program [p,i])) | p <- setting] 0
in maximum (map outcome settings)
part2 :: [Integer] -> Integer
part2 program =
let settings = permutations [5..9]
outcome setting =
let initConts = [let Left (cont, _) = runInterruptible program [p] in cont
| p <- setting]
generation conts firstInp =
let (output, results) = foldProduce (\inp cont ->
let res = runContinue cont [inp]
in (res, head (either snd snd res)))
firstInp conts
in case last results of
Right (_, _) -> output
Left (_, _) -> generation (map (fst . fromLeft undefined) results) output
in generation initConts 0
in maximum (map outcome settings)
foldProduce :: (s -> a -> (b, s)) -> s -> [a] -> (s, [b])
foldProduce _ s [] = (s, [])
foldProduce f s (x:xs) = let (y, s') = f s x in fmap (y :) (foldProduce f s' xs)
main :: IO ()
main = do
program <- parse . head <$> getInput 7
print (part1 program)
print (part2 program)
|
