61 lines
2.8 KiB
Haskell
61 lines
2.8 KiB
Haskell
|
import Data.List.Split
|
||
|
|
import Data.List
|
||
|
|
import Data.Char as Char
|
||
|
|
import Linear.V3
|
||
|
|
import Debug.Trace as T
|
||
|
|
|
||
|
|
main = do
|
||
|
|
reactions <- map getReaction <$> lines <$> getContents
|
||
|
|
let amout = getConstruction reactions (1,"A")
|
||
|
|
let ore = getOre [(1,"FUEL")] [] reactions
|
||
|
|
let result = foldl (++)[] (map(\(amount,xs) -> map(\(a,e) -> ((amount * a),e)) xs) (map (getConstructionRest reactions) ore))
|
||
|
|
let sumResult = foldl combineNeeds [] result
|
||
|
|
putStrLn(show $ head reactions)
|
||
|
|
putStrLn(show ore)
|
||
|
|
putStrLn(show sumResult)
|
||
|
|
data Reaction = Reaction { input :: [(Int, String)],
|
||
|
|
output :: (Int,String)
|
||
|
|
} deriving Show
|
||
|
|
|
||
|
|
getReaction :: String -> Reaction
|
||
|
|
getReaction input = Reaction left right
|
||
|
|
where split = splitOn " => " input
|
||
|
|
left = map getElements (splitOn ", " (split!!0))
|
||
|
|
right = getElements $ split!!1
|
||
|
|
|
||
|
|
getElements :: String -> (Int,String)
|
||
|
|
getElements input = (amount,element)
|
||
|
|
where split = splitOn " " input
|
||
|
|
amount = read $ (split!!0)
|
||
|
|
element = split!!1
|
||
|
|
|
||
|
|
getNextStep :: [Reaction] -> Reaction -> [Reaction]
|
||
|
|
getNextStep xs (Reaction left right) = filter(\(Reaction i o) -> elem (snd o) reactElem ) xs
|
||
|
|
where reactElem = map(\(a,b) -> b) left
|
||
|
|
|
||
|
|
getConstruction :: [Reaction] -> (Int,String) -> (Int,[(Int, String)])
|
||
|
|
getConstruction reactions (amount,elem)
|
||
|
|
| length reaction > 0 = (div amount ( fst $ output $ head $ reaction),(input $ head $ reaction))
|
||
|
|
| otherwise = ((div amount ( fst $ output $ head $ reaction)) + 1,(input $ head $ reaction))
|
||
|
|
where reaction = filter(\(Reaction i o) -> ((snd o) == elem) && (mod amount (fst o) == 0)) reactions
|
||
|
|
|
||
|
|
|
||
|
|
getConstructionRest :: [Reaction] -> (Int,String) -> (Int,[(Int, String)])
|
||
|
|
getConstructionRest reactions (amount, elem)
|
||
|
|
| elem == "ORE" = (1,[(amount,elem)])
|
||
|
|
| otherwise = ((div amount ( fst $ output $ head $ reaction)) + 1,(input $ head $ reaction))
|
||
|
|
where reaction = filter(\(Reaction i o) -> ((snd o) == elem)) reactions
|
||
|
|
|
||
|
|
getOre :: [(Int,String)] -> [(Int,String)] -> [Reaction] -> [(Int,String)]
|
||
|
|
getOre needs oldNeeds reactions
|
||
|
|
| needs == oldNeeds = newNeeds
|
||
|
|
| otherwise = getOre (T.traceShowId(newNeeds)) needs reactions
|
||
|
|
where newNeeds =foldl combineNeeds [] ( foldl (++)[] (map(\(amount,xs) -> map(\(a,e) -> ((amount * a),e)) xs) construction))
|
||
|
|
construction = map (getConstruction reactions) needs
|
||
|
|
|
||
|
|
combineNeeds :: [(Int,String)] -> (Int,String) -> [(Int,String)]
|
||
|
|
combineNeeds xs (amt, elem)
|
||
|
|
| null oldVal = xs ++ [(amt,elem)]
|
||
|
|
| otherwise = (xs \\ oldVal) ++ [((amt + (fst (head oldVal))),elem)]
|
||
|
|
where oldVal = filter(\(a,e) -> e == elem) xs
|