From 9da7f985908df182355b04f7564f2d1109a68a1e Mon Sep 17 00:00:00 2001
From: Arranun <karliosos65@googlemail.com>
Date: Sat, 14 Dec 2019 17:15:07 +0100
Subject: [PATCH] Day14: All test are working

---
 day14.hs | 38 +++++++++++++++++++++++++++++++-------
 1 file changed, 31 insertions(+), 7 deletions(-)

diff --git a/day14.hs b/day14.hs
index bae2908..e4a96e6 100644
--- a/day14.hs
+++ b/day14.hs
@@ -8,11 +8,13 @@ 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
+      let result = getRestOre ore [] reactions
+      let test = part1 [(1,"FUEL")] [] reactions
       putStrLn(show $ head reactions)
-      putStrLn(show ore)
-      putStrLn(show sumResult)
+      --putStrLn(show ore)
+      putStrLn(show result)
+      putStrLn(show test)
+      --putStrLn(show sumResult1)
 data Reaction = Reaction { input :: [(Int, String)],
                            output :: (Int,String)
                          } deriving Show
@@ -35,8 +37,9 @@ getNextStep xs (Reaction left right) = filter(\(Reaction i o) -> elem (snd  o)
     
 getConstruction ::  [Reaction] -> (Int,String) -> (Int,[(Int, String)])
 getConstruction reactions (amount,elem) 
+  |  elem == "ORE" = (1,[(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))
+  |  otherwise = (1,[(amount,elem)])
                    where reaction =  filter(\(Reaction i o) -> ((snd  o) == elem) && (mod amount (fst o) == 0)) reactions
       
 
@@ -44,17 +47,38 @@ 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
+                where reaction =  filter(\(Reaction i o) -> ((snd  o) == elem) ) reactions
+      
+part1 :: [(Int,String)] -> [(Int,String)] -> [Reaction] ->  [(Int,String)]  
+part1 needs oldNeeds reactions 
+         |needs == oldNeeds = needs
+         | otherwise = part1 newNeeds needs reactions
+         where newNeeds = getRestOre fullNeeds [] reactions
+               fullNeeds = getOre needs [] reactions
 
 getOre :: [(Int,String)] -> [(Int,String)] -> [Reaction] ->  [(Int,String)]
 getOre needs oldNeeds reactions
          | needs == oldNeeds = newNeeds 
-         | otherwise = getOre  (T.traceShowId(newNeeds)) needs reactions  
+         | otherwise = getOre  (newNeeds) needs reactions  
          where newNeeds =foldl combineNeeds [] ( foldl (++)[] (map(\(amount,xs) -> map(\(a,e) -> ((amount * a),e)) xs) construction))
                construction = map (getConstruction reactions) needs
 
+
+getRestOre :: [(Int,String)] -> [(Int,String)] -> [Reaction] -> [(Int,String)]
+getRestOre needs oldNeeds reactions = sumNeeds
+        where  pureNeeds = getPureNeeds needs reactions
+               pureRest = foldl (++)[] (map(\(amount,xs) -> map(\(a,e) -> ((amount * a),e)) xs) (map (getConstructionRest reactions) pureNeeds))
+               sumNeeds = foldl combineNeeds [] ((needs \\ pureNeeds) ++ pureRest)  
+
 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
+      
+getPureNeeds :: [(Int,String)] -> [Reaction] -> [(Int,String)]
+getPureNeeds needs reactions = filter(\(a,e) ->  notElem e impureElements ) needs 
+                  where   impureElements = map(\(a,e) -> e) ( (foldl (++) [] ( map(\(Reaction i o) -> i)reactionList )))
+                          reactionList = (filter(\(Reaction i o) -> elem (snd o) elements) reactions)
+                          elements = (map(\(a,e) -> e) needs)
+