|
| 1 | +{- |
| 2 | +In the 20×20 grid below, four numbers along a diagonal line have been marked in |
| 3 | +red. |
| 4 | +
|
| 5 | +08 02 22 97 38 15 00 40 00 75 04 05 07 78 52 12 50 77 91 08 |
| 6 | +49 49 99 40 17 81 18 57 60 87 17 40 98 43 69 48 04 56 62 00 |
| 7 | +81 49 31 73 55 79 14 29 93 71 40 67 53 88 30 03 49 13 36 65 |
| 8 | +52 70 95 23 04 60 11 42 69 24 68 56 01 32 56 71 37 02 36 91 |
| 9 | +22 31 16 71 51 67 63 89 41 92 36 54 22 40 40 28 66 33 13 80 |
| 10 | +24 47 32 60 99 03 45 02 44 75 33 53 78 36 84 20 35 17 12 50 |
| 11 | +32 98 81 28 64 23 67 10[26]38 40 67 59 54 70 66 18 38 64 70 |
| 12 | +67 26 20 68 02 62 12 20 95[63]94 39 63 08 40 91 66 49 94 21 |
| 13 | +24 55 58 05 66 73 99 26 97 17[78]78 96 83 14 88 34 89 63 72 |
| 14 | +21 36 23 09 75 00 76 44 20 45 35[14]00 61 33 97 34 31 33 95 |
| 15 | +78 17 53 28 22 75 31 67 15 94 03 80 04 62 16 14 09 53 56 92 |
| 16 | +16 39 05 42 96 35 31 47 55 58 88 24 00 17 54 24 36 29 85 57 |
| 17 | +86 56 00 48 35 71 89 07 05 44 44 37 44 60 21 58 51 54 17 58 |
| 18 | +19 80 81 68 05 94 47 69 28 73 92 13 86 52 17 77 04 89 55 40 |
| 19 | +04 52 08 83 97 35 99 16 07 97 57 32 16 26 26 79 33 27 98 66 |
| 20 | +88 36 68 87 57 62 20 72 03 46 33 67 46 55 12 32 63 93 53 69 |
| 21 | +04 42 16 73 38 25 39 11 24 94 72 18 08 46 29 32 40 62 76 36 |
| 22 | +20 69 36 41 72 30 23 88 34 62 99 69 82 67 59 85 74 04 36 16 |
| 23 | +20 73 35 29 78 31 90 01 74 31 49 71 48 86 81 16 23 57 05 54 |
| 24 | +01 70 54 71 83 51 54 69 16 92 33 48 61 43 52 01 89 19 67 48 |
| 25 | +
|
| 26 | +The product of these numbers is 26 × 63 × 78 × 14 = 1788696. |
| 27 | +
|
| 28 | +What is the greatest product of four adjacent numbers in the same direction |
| 29 | +(up, down, left, right, or diagonally) in the 20×20 grid? |
| 30 | +-} |
| 31 | + |
| 32 | +-- Get rows |
| 33 | +-- Get columns |
| 34 | +-- Get windowed rows, columns, diagonals |
| 35 | +-- Calculate product, take maximum |
| 36 | + |
| 37 | +import Data.List |
| 38 | +import Data.Array |
| 39 | +import Text.Printf |
| 40 | +import Control.Exception |
| 41 | +import System.CPUTime |
| 42 | +import System.Environment |
| 43 | + |
| 44 | +main :: IO () |
| 45 | +time :: IO t -> IO t |
| 46 | +time a = do |
| 47 | + start <- getCPUTime |
| 48 | + v <- a |
| 49 | + end <- getCPUTime |
| 50 | + let diff = (fromIntegral (end - start)) / (10^12) |
| 51 | + printf "Computation time: %0.3f sec\n" (diff :: Double) |
| 52 | + return v |
| 53 | + |
| 54 | +-- |
| 55 | +window :: Int -> [Int] -> [[Int]] |
| 56 | +window n list = |
| 57 | + case list of |
| 58 | + [] -> [] |
| 59 | + x:xs -> |
| 60 | + if length list >= n |
| 61 | + then ( x : (take (n-1) xs) ) : window n xs |
| 62 | + else window n xs |
| 63 | + |
| 64 | +hasZero :: [Int] -> Bool |
| 65 | +hasZero [] = True |
| 66 | +hasZero (x:xs) = if x == 0 then True else hasZero xs |
| 67 | +-- |
| 68 | + |
| 69 | +-- solution :: Integer -> IO () |
| 70 | +-- solution n = putStrLn $ window n |
| 71 | +-- getRows = (map (map read) $ map words $ lines contents :: [[Int]]) |
| 72 | + |
| 73 | +main = do |
| 74 | + contents <- readFile "./number.txt" |
| 75 | + let rows = map (map read) $ map words $ lines contents :: [[Int]] |
| 76 | + let cols = transpose rows |
| 77 | + let diags = filter hasZero $ filter (((==4) . length)) $ concatMap (transpose . zipWith drop [0..3]) (tails rows) |
| 78 | + let diags2 = filter hasZero $ filter ((==4) . length) $ concatMap (transpose . zipWith drop [0..3]) (tails $ reverse rows) |
| 79 | + let mappedRows = filter hasZero $ concatMap (window 4) rows |
| 80 | + let mappedCols = filter hasZero $ concatMap (window 4) cols |
| 81 | + let all = diags ++ diags2 ++ mappedRows ++ mappedCols |
| 82 | + let solution = maximum $ map product all |
| 83 | + putStrLn "Starting..." |
| 84 | + putStrLn $ show solution |
| 85 | + -- time $ solution number |
| 86 | + putStrLn "Done." |
| 87 | + |
0 commit comments