summaryrefslogtreecommitdiff log msg author committer range
blob: 5e29679425c89226b0bcdfba7e7d95315c8f3617 (plain)
 ```1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 ``` ``````{-# LANGUAGE ParallelArrays #-} {-# OPTIONS -fvectorise -XParallelListComp #-} module DiophantineVect (solution3) where import Data.Array.Parallel import Data.Array.Parallel.Prelude.Int as I import qualified Prelude as P solution3' = let pow x i = productP (replicateP i x) primes = [: 2,3,5,7,11,13,17,19,23,29,31,37,41,43,47,53,59,61,67,71,73 :] a `cutTo` b = sliceP 0 (lengthP b) a sumpri xx = productP [: pow p x | p <- primes `cutTo` xx | x <- xx :] distinct xx = productP [: x I.+ 1 | x <- xx :] series :: [:Int:] -> Int -> [:[:Int:]:] series xs n | n == 1 = [: [: 0 :] :] | otherwise = [: [: x :] +:+ ps | x <- xs , ps <- series (I.enumFromToP 0 x) (n I.- 1) :] prob x y = let xx = [: (sumpri m ,m) | m <- series (I.enumFromToP 1 3) x , distinct [: x I.* 2 | x <- m :] > y :] i = minIndexP [: a | (a, b) <- xx :] in xx !: i in prob 5 200 solution3 :: (Int, PArray Int) {-# NOINLINE solution3 #-} solution3 = let (i, is) = solution3' in (i, toPArrayP is) ``````