Google Treasure Hunt primes question
In the comments to my post on the Google treasure hunt Andre asked me to put up my solution for the fourth problem. The problems were personalised and here is mine:
Find the smallest number that can be expressed as
the sum of 3 consecutive prime numbers,
the sum of 29 consecutive prime numbers,
the sum of 373 consecutive prime numbers,
the sum of 665 consecutive prime numbers,
and is itself a prime number.
The idea behind my solution is that if I have four lists of the sums, and a list of primes, then the solution to the problem is the smallest member of the intersection of all five lists.
I saved the problem of finding primes to the last and instead concentrated on summing up lists of integers in a convenient way. I ended up with the following solution to that:
sumN :: Int -> [Integer] sumN n = map fst sumi where sumi = tail $ iterate (sumn) (0, primes) sumn (r, xs) = (sum $ take n xs, tail xs)
The real work happens in the inner functions. sumn picks out as many numbers as I want and sums them up, the reason for taking and returning a tuple is that sumi can use iterate to calculate the sequence of sums. The first item in the result of iterate a is a, so sumi drops the first one. This lets me create the needed lists of sums:
sum3 = sumN 3 sum29 = sumN 29 sum373 = sumN 373 sum665 = sumN 665
After having a quick look at the implementation of Data.List.intersect I realised it probably would be a bit too slow. Instead I wrote one that exploits that all the involved lists are ordered:
intersection a@(x:xs) b@(y:ys) | x == y = x : intersection xs ys | x < y = intersection (dropWhile (< y) xs) b | x > y = intersection a (dropWhile (x >) ys)
After convincing myself that this code did what I wanted (basically through playing in GHCi using an incorrect definition of primes, primes = [1..]) I turned to the problem of finding the primes. Of course someone else had already written the Haskell code to find primes. That code fitted perfectly with mine since it produces an infinate list of Integers. Brilliant! 
Finally here’s the main that allows me to compile the code and get the answer quickly (it’s slightly modified for the medium, basically I wrote it all on one, long line):
main :: IO () main = let i1 = (primes `intersection` sum665) i2 = (sum3 `intersection` sum373) s = i1 `intersection` i2 `intersection` sum29 in print $ head s
Compiled, it runs in about 5.5 seconds on my desktop machine.
![[Digg]](http://therning.org/magnus/wp-content/plugins/bookmarkify/digg.png)
![[Reddit]](http://therning.org/magnus/wp-content/plugins/bookmarkify/reddit.png)
foobar:
The following looks nicer to me:
sumN 1 = primes
8 June 2008, 10:00 amsumN n = [x+y | (x, y) <- zip (tail $ sumN (n-1)) primes]
Andre:
Ah, I see. My problem was the naive implementation of the sieve. The optimized version runs much faster.
My intersection-function is similar, although it finds elements which are in each passed (infinite) list. Here’s my implementation:
intersect :: (Eq a, Ord a) => [[a]] -> [a]
intersect [] = []
intersect xs | or (map null xs) = []
| and $ map ((== x) . head) xs = x : (intersect $ map tail xs)
| otherwise = intersect $ map (dropWhile (< max)) xs
where max = let comp (x:) (y:) = compare x y
in head $ maximumBy comp xs
x = head $ head xs
(Sorry for the formatting.)
Therefore I’ve a more general function for finding the prime which fulfills the required conditions:
prime :: [Int] -> Integer
prime = head . intersect . (primes:) . map (`calc` primes)
Whereas calc is semantically equivalent to your sumN-function.
Using your test-data and the optimized sieve my program needs circa 10 seconds for finding the solution. Great work of yours!
8 June 2008, 12:33 pmRussell O'Connor:
This is almost the same as my code, except I wrote
8 June 2008, 3:50 pmsumNassumN xs = map (sum . take n) (tails primes)Peter:
Thanks for showing us what you did!
I really like foobar’s definition of sumN — much better than the one I came up with — although I’d have used zipWith, like this:
sumN 1 = primessumN (n+1) = zipWith (+) primes $ tail $ sumN n
For extra prettiness, notice that primes == sumN 1 and that Andre’s version of intersect can be derived from Magnus’ version of intersect using foldl1:
main = print $ head $ foldl1 intersect [sumN n primes | n <- [1, 3, 29, 373, 665]]
9 June 2008, 1:50 amPeter:
Oops. There’s a typo in my previous comment. It should be:
9 June 2008, 1:53 ammain = print $ head $ foldl1 intersect [sumN n | n <- [1, 3, 29, 373, 665]]Magnus:
@Russel,
That is a beautiful definition of
sumN! I wasn’t aware of the existance of thetailsfunction before.@Peter,
Thanks for pointing that out, it does make it a bit more beautiful and readable.
9 June 2008, 8:37 amSamuel:
Interesting how we all used pretty much the same solution.
I used the following definition of sumN (equivalent to Russel’s function). Should be much faster, since it doesn’t need to iterate through 665 primes for each element.
My definition of primes, btw, is the following. Slower than the fastest one on the quoted page, but still pretty fast and elegant.
9 June 2008, 10:45 amNikolay:
I’ve used such definition of sums with stream-fusion:
That should prevent much GC, because all we need is primes list, all other is shifting sums.
10 June 2008, 12:12 pmciju:
Here’s another solution. Runs in around 0.2 seconds, although using C.
http://ciju.wordpress.com/2008/06/03/google-treasure-hunt/
The method should work in haskell also. If you try it, please do let me know how much speedup u get, if any.
10 June 2008, 3:38 pm