X-Git-Url: https://git.njae.me.uk/?a=blobdiff_plain;f=advent11.hs;h=33bfe3c7cc2e788c03cfdc65bc46984179a0494c;hb=a792a40d6c4b7cf09faa9fb5d04d27f71d2a49a7;hp=58215fd6da22ec671ee05f3b22d21d66ea734ca3;hpb=5fd57ba9da53a1ee824ea4153090b40f9dd15ebd;p=advent-of-code-16.git

diff --git a/advent11.hs b/advent11.hs
index 58215fd..33bfe3c 100644
--- a/advent11.hs
+++ b/advent11.hs
@@ -48,25 +48,37 @@ main = do
 
 
 part1 :: IO ()
--- part1 = print $ length $ init $ extractJust $ aStar [[buildingTest]] []
-part1 = print $ length $ init $ extractJust $ aStar [[building1]] []
+-- part1 = print $ length $ init $ extractJust $ hillClimb [[buildingTest]] []
+part1 = print $ length $ init $ extractJust $ hillClimb [[building1]] []
+-- part1 = print $ length $ init $ extractJust $ aStar [[building1]] []
 
 part2 :: IO ()
-part2 = print $ length $ init $ extractJust $ aStar [[building2]] []
+part2 = print $ length $ init $ extractJust $ hillClimb [[building2]] []
 
 
 extractJust :: Maybe [a] -> [a]
 extractJust Nothing = []
 extractJust (Just x) = x
 
+hillClimb :: [[Building]] -> [Building] -> Maybe [Building]
+hillClimb [] _ = Nothing
+hillClimb (currentTrail:trails) closed = 
+    if isGoal (head currentTrail) then Just currentTrail
+    else hillClimb newAgenda ((head currentTrail): closed) 
+    where newAgenda = 
+            sortBy (\t1 t2 -> (head t1) `compare` (head t2)) $ 
+            trails ++ (candidates currentTrail closed)
+
 aStar :: [[Building]] -> [Building] -> Maybe [Building]
 aStar [] _ = Nothing
 aStar (currentTrail:trails) closed = 
     if isGoal (head currentTrail) then Just currentTrail
     else aStar newAgenda ((head currentTrail): closed) 
     where newAgenda = 
-            sortBy (\t1 t2 -> (head t1) `compare` (head t2)) $ 
+            sortBy (\t1 t2 -> (trailCost t1) `compare` (trailCost t2)) $ 
             trails ++ (candidates currentTrail closed)
+          trailCost t = estimateCost (head t) + length t - 1
+
 
 candidates :: [Building] -> [Building] -> [[Building]]
 candidates currentTrail closed = newCandidates