Problem 68

Project Euler

http://projecteuler.net/index.php?section=problems&id=68
とりあえず一般化してmagic nを求められるようにした。
始めはsの範囲を間違えていて正しい解が出なかった。
ちなみに重複を除くのは面倒なのでやっていない。

import Data.List
import Data.Ord
import Data.Char
import Control.Monad
magic n = do 
  s <- [6..(6*n-3)]
  b0 <- [1..2*n]
  magic' s b0 b0 $[1..2*n]\\[b0]>>=return

magic' s b0 b [a] = [[[a,b,b0]]|s==a+b+b0]
magic' s b0 b xs = do
  a <- xs
  let c = s-a-b
  guard.elem c$xs\\[a]
  magic' s b0 c (xs\\[a,c])>>=return.([a,b,c]:)

normal xs = let m = minimumBy (comparing head) xs
                (ys,zs) = break (==m) xs
            in zs++ys
main =print.maximum.filter((==16).length). map (concatMap show.concat).nub.map normal .magic$5