32 lines
934 B
Haskell
Executable file
32 lines
934 B
Haskell
Executable file
#!/usr/bin/env runghc
|
|
module Day7.Part1 where
|
|
|
|
import Day7
|
|
import Data.Array
|
|
import Data.Graph
|
|
import Data.Maybe
|
|
|
|
main = interact (show . solve "shiny gold" . parse rules)
|
|
|
|
-- solve myBag rs = (+) (-1) $ forestSize $ dfs graph [vertex]
|
|
-- solve myBag rs = length $ keys -- filter (path graph vertex . fromJust . vertexFromKey) keys
|
|
solve myBag rs = graph
|
|
where keys = filter (/= myBag) $ map ruleToKey rs
|
|
(graph, keyFromVertex, vertexFromKey) = graphFromEdges . map ruleToEdge $ rs
|
|
Just vertex = vertexFromKey myBag
|
|
|
|
|
|
ruleToKey = name . container
|
|
|
|
forestSize [] = 0
|
|
forestSize ((Node _ branches):siblings) = nodeSum + siblingSum + branchSum
|
|
where
|
|
nodeSum = 1
|
|
siblingSum = forestSize siblings
|
|
branchSum = forestSize branches
|
|
|
|
ruleToEdge :: Rule -> (String, String, [String])
|
|
ruleToEdge r = (node, node, edges)
|
|
where
|
|
node = name . container $ r
|
|
edges = map (name . item) . contents $ r
|