a good first step for pb 68

068.hs

@@ -0,0 +1,112 @@
+-- Consider the following "magic" 3-gon ring, filled with the numbers 1 to 6, and each line adding to nine.
+--
+-- Working clockwise, and starting from the group of three with the numerically lowest external node (4,3,2 in this example), each solution can be described uniquely. For example, the above solution can be described by the set: 4,3,2; 6,2,1; 5,1,3.
+--
+-- It is possible to complete the ring with four different totals: 9, 10, 11, and 12. There are eight solutions in total.
+--             Total   Solution Set
+--             9   4,2,3; 5,3,1; 6,1,2
+--             9   4,3,2; 6,2,1; 5,1,3
+--             10  2,3,5; 4,5,1; 6,1,3
+--             10  2,5,3; 6,3,1; 4,1,5
+--             11  1,4,6; 3,6,2; 5,2,4
+--             11  1,6,4; 5,4,2; 3,2,6
+--             12  1,5,6; 2,6,4; 3,4,5
+--             12  1,6,5; 3,5,4; 2,4,6
+--
+-- By concatenating each group it is possible to form 9-digit strings; the maximum string for a 3-gon ring is 432621513.
+--
+-- Using the numbers 1 to 10, and depending on arrangements, it is possible to form 16- and 17-digit strings. What is the maximum 16-digit string for a "magic" 5-gon ring?
+
+-- SOLUTION in English
+{-
+
+For each digit from 1 to 10:
+  put the digit in the current partially filled NGon
+
+ -}
+import Data.List
+import Debug.Trace
+
+
+-- For testing
+gonSize = 3
+magic = 9
+
+-- gonSize = 5
+-- magic = 16
+
+data Choice = Choice [Int]
+
+safeIndex s l i = if (length l<i+1) then trace ("ERROR (" ++ s ++ "): " ++ show l ++ "(" ++ show i ++ ")") l!!i else l!!i
+
+instance Show Choice where
+  show (Choice l)= str ++ show l
+    where
+      n = length l
+      nbLines = if n<2*gonSize then (n-1) `div` 2 else gonSize
+      str = concatMap show $ (take 3 l) ++ (concatMap (line l) [2..nbLines])
+      line l n=map (safeIndex "show" l) [b+1,b,lastelem]
+        where
+          b=2*(n-1)
+          lastelem=if n == gonSize then 1 else b+2
+
+
+class RAS a where
+  at :: a -> Int -> Int
+  nbChoices :: a -> Int
+  remove :: a -> Int -> a
+  add :: a -> Int -> a
+  loop :: (Int -> b) -> a -> [b]
+
+instance RAS Choice where
+  at (Choice l) i = safeIndex "at" l i
+  nbChoices (Choice l) = length l
+  remove (Choice l) i = Choice $ filter (\x -> x/=i) l
+  add (Choice l) e = Choice (l++[e])
+  loop f (Choice l) = map f l
+
+testPartialGon :: Choice -> Bool
+testPartialGon c =
+  let
+    n = nbChoices c
+    nbLines = if n<2*gonSize then (n-1) `div` 2 else gonSize
+  in
+  all (testLine c magic) [1..nbLines]
+
+testLine :: Choice -> Int -> Int -> Bool
+testLine c val n =
+  let
+    b=max 0 2*(n-1)
+    lastelem=if n == gonSize then 1 else b+2
+    line=[b,b+1,lastelem]
+  in
+    (==val) . sum . map (at c) $ line
+
+allTests :: [Choice]
+allTests = testWith (Choice []) (Choice [1..2*gonSize])
+
+testWith ::    Choice   -- choosen
+            -> Choice   -- left choices
+            -> [Choice] -- successful choices
+testWith c lc =
+  if testPartialGon c
+  then if nbChoices lc == 0
+       then [c]
+       else concat $ loop newTest lc
+  else []
+    where
+      len = nbChoices c
+      newTest x = if len>=3 && (len `rem` 2 == 1) && x<at c 0
+                    then trace (show c ++ " len: " ++ show len ++ " " ++ show x ++ "<" ++ show (at c 0)) []
+                    else testWith (add c x) (remove lc x)
+
+main :: IO ()
+main = do
+  print $ Choice [4,3,2,6,1,5]
+  putStr "testPartialGon $ Choice [4,3,2,6,1,5]: "
+  print $ testPartialGon $ Choice [4,3,2,6,1,5]
+  putStr "testPartialGon $ Choice [6,2,1,3,5,4]: "
+  print $ testPartialGon $ Choice [6,2,1,3,5,4]
+  print $ Choice [6,2,1,3,5,4]
+  putStr "allTest: "
+  print $ allTests