Realised I was doing hillclimbing search, not A*

[advent-of-code-16.git] / advent11.hs

diff --git a/advent11.hs b/advent11.hs
index af048de2f7090c05d09ebd1a68b9a058eb386396..5fd4e40b784658a12580906efbbb2cb5774d336f 100644 (file)
--- a/advent11.hs
+++ b/advent11.hs
@@ -48,22 +48,22 @@ 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]] []
 
 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
 
-aStar :: [[Building]] -> [Building] -> Maybe [Building]
-aStar [] _ = Nothing
-aStar (currentTrail:trails) closed = 
+hillClimb :: [[Building]] -> [Building] -> Maybe [Building]
+hillClimb [] _ = Nothing
+hillClimb (currentTrail:trails) closed = 
     if isGoal (head currentTrail) then Just currentTrail
-    else aStar newAgenda ((head currentTrail): closed) 
+    else hillClimb newAgenda ((head currentTrail): closed) 
     where newAgenda = 
             sortBy (\t1 t2 -> (head t1) `compare` (head t2)) $ 
             trails ++ (candidates currentTrail closed)
@@ -110,10 +110,8 @@ legalSuccessors :: [Building] -> [Building]
 legalSuccessors = filter (isLegal)
 
 updateBuilding :: Int -> [Floor] -> Int -> [Item] -> Building
--- updateBuilding f floors _ _ = Building f floors
 updateBuilding oldF oldFloors newF items = Building newF newFloors
-    where numberedFloors = zip [0..] oldFloors
-          newFloors = map (updateFloor) numberedFloors
+    where newFloors = map (updateFloor) $ zip [0..] oldFloors
           updateFloor (f, fl) 
             | f == oldF = sort $ fl \\ items
             | f == newF = sort $ items ++ fl