HaskellでProjectEulerを。11〜

Haskell ProjectEuler

そろそろ続きを書こう、と思っていたら下書き書いてそのままだったProjectEulerの11問目から。
下書きのタイムスタンプを見ると2009-09-08 02:34。
どうやら深夜に書き始めて、途中で気力が無くなったようだ。
さすがに今から加筆修正するのもアレだし、下書きママ貼り付けるのもありかな、とか思ったので、そのまま張ります。
文中に4ヶ月前、と書いてありますが、9月8日での、と適当に補完して下さい。
書きかけすいませぬ。
14問目のコラッツ数列まで。


↓ここから9月8日 現在↓


前回から実に4ヶ月以上のブランクを置いてやっと続き。
HaskellでProjectEulerに挑戦していこうと思います。
今回は11番から。

まずはLib.hs

前回も張ったのですが、多少手を加えたので。
といっても素数列生成のところで書いたのだけですが。

module Lib where

isPerfectNumber :: (Integral a) => a -> Bool
isPerfectNumber n = (n ==) $ sum $ properDivisors n

isDeficientNumber :: (Integral a) => a -> Bool
isDeficientNumber n = (n >) $ sum $ properDivisors n

isAbundantNumber :: (Integral a) => a -> Bool
isAbundantNumber n = (n <) $ sum $ properDivisors n

diff :: (Ord a, Eq a) => [a] -> [a] -> [a]
diff xxs@(x:xs) yys@(y:ys)
	| x > y		= diff xxs ys
	| x < y		= x:diff xs yys
	| otherwise	= diff xs ys
diff [] _		= []
diff xxs []		= xxs

merge :: (Ord a) => [a] -> [a] -> [a]
merge xxs@(x:xs) yys@(y:ys)
	| x < y		= x:merge xs yys
	| x > y		= y:merge xxs ys
	| otherwise	= x:merge xs ys
merge [] yys		= yys
merge xxs []		= xxs

fib :: (Num a) => a -> a -> [a]
fib x1 x2 = x1:x2:zipWith (+) (fib x1 x2) (tail (fib x1 x2))
--fib@(_:tfib) = 1:2:zipWith (+) fib tfib

properDivisors :: (Integral a) => a -> [a]
properDivisors num = 1:divisors' num [2..num]
	where
		divisors' _ []			= []
		divisors' n (x:xs)
			| n < x * x		= []
			| n == x * x		= [x]
			| n `mod` x == 0	= x:(n `div` x):divisors' n xs
			| otherwise		= divisors' n xs

divisors :: (Integral a) => a -> [a]
divisors 1 = [1]
divisors num = divisors' num [1..num]
	where
		divisors' _ []			= []
		divisors' n (x:xs)
			| n < x * x		= []
			| n `mod` x == 0	= x:(n `div` x):divisors' n xs
			| otherwise		= divisors' n xs


multiples :: (Num a) => a -> [a]
multiples x = iterate (+x) x

isqrt :: (Integral a, Integral b) => a -> b
isqrt = ceiling . (sqrt :: Double -> Double) . fromIntegral

primes :: (Integral a) => [a]
primes = sieve [2..]

sieve :: (Num a, Ord a) => [a] -> [a]
sieve = sieve' [2]

sieve' :: (Num a, Ord a) => [a] -> [a] -> [a]
sieve' [] [] = []
sieve' [] (x:xs) = sieve' [x] xs
sieve' (p:ps) xxs =
	let (p1,p2) = span (< p^(2::Int)) ((xxs `diff`) $ tail $ multiples p) in
	(p1 ++) $ sieve' (ps ++ p1) p2

-- primes :: (Num a, Ord a) => [a]
-- primes = 2:primes' (iterate (+2) 3)
-- 	where
-- 		primes' (x:xs)	= x:(primes' $ xs `diff` (iterate (+x) x))
-- 		primes' _	= []
-- primes :: (Integral a) => [a]
-- primes = primes' (2:[3,5..])
-- 	where
-- 		primes' (x:xs)	= x:(primes' $ filter ((/=0).(`mod`x)) xs)
-- 		primes' _	= []

factors :: (Integral a) => a -> [a]
factors = (`factors'` primes)
	where
		factors' n xxs@(x:xs)
			| n < x * x		= [n]
			| n `mod` x == 0	= x:factors' (n `div` x) xxs
			| otherwise		= factors' n xs
		factors' _ _			= []

foldl' :: (a -> b -> a) -> a -> [b] -> a
foldl' _ a [] = a
foldl' f a (x:xs) = (foldl' f $! f a x) xs

foldl1' :: (a -> a -> a) -> [a] -> a
foldl1' f (x:xs) = foldl' f x xs
foldl1' _ _ = error "null list."

あいかわらずです(何が)

Project Euler 11


08 02 22 97 38 15 00 40 00 75 04 05 07 78 52 12 50 77 91 08
49 49 99 40 17 81 18 57 60 87 17 40 98 43 69 48 04 56 62 00
81 49 31 73 55 79 14 29 93 71 40 67 53 88 30 03 49 13 36 65
52 70 95 23 04 60 11 42 69 24 68 56 01 32 56 71 37 02 36 91
22 31 16 71 51 67 63 89 41 92 36 54 22 40 40 28 66 33 13 80
24 47 32 60 99 03 45 02 44 75 33 53 78 36 84 20 35 17 12 50
32 98 81 28 64 23 67 10 26 38 40 67 59 54 70 66 18 38 64 70
67 26 20 68 02 62 12 20 95 63 94 39 63 08 40 91 66 49 94 21
24 55 58 05 66 73 99 26 97 17 78 78 96 83 14 88 34 89 63 72
21 36 23 09 75 00 76 44 20 45 35 14 00 61 33 97 34 31 33 95
78 17 53 28 22 75 31 67 15 94 03 80 04 62 16 14 09 53 56 92
16 39 05 42 96 35 31 47 55 58 88 24 00 17 54 24 36 29 85 57
86 56 00 48 35 71 89 07 05 44 44 37 44 60 21 58 51 54 17 58
19 80 81 68 05 94 47 69 28 73 92 13 86 52 17 77 04 89 55 40
04 52 08 83 97 35 99 16 07 97 57 32 16 26 26 79 33 27 98 66
88 36 68 87 57 62 20 72 03 46 33 67 46 55 12 32 63 93 53 69
04 42 16 73 38 25 39 11 24 94 72 18 08 46 29 32 40 62 76 36
20 69 36 41 72 30 23 88 34 62 99 69 82 67 59 85 74 04 36 16
20 73 35 29 78 31 90 01 74 31 49 71 48 86 81 16 23 57 05 54
01 70 54 71 83 51 54 69 16 92 33 48 61 43 52 01 89 19 67 48

上の 20 × 20 の数字のなか、赤くマークされた数字の積は 26 × 63 × 78 × 14 = 1788696 となる。

上下左右斜めのいずれかの方向で連続する4つの数字の積のうち最大のものを求めよ。

なにやら既視感を覚えつつもLet's try la lai!
Problem8では連続する5つ、と同じ行だけを見ればよかったのが、今回は縦横斜め。
縦はtranspose(:: [ [ a ] ] -> [ [ a ] ])を使えばok。
斜めは各行それぞれの先頭に、1行目は0個、2行目は1個、3行目は2個...とインデントをいれてtransposeすればok。
という方針で行ってみました。
んー。美しくないなぁ。なにかよい方法なないものか。

import List

main :: IO ()
main = do
	str <- readFile "11.txt"
	let nums = map (map (read::String->Integer) . words) $ lines str in do
		print $ maximum $ concat $ product1 nums ++ product2 nums ++ product3 nums ++ product4 nums

product1 :: (Num a) => [[a]] -> [[a]]
product1 = map product1'
	where
	product1' xxs@(_:xs@(_:_:_:_))	= (product $ take 4 xxs):product1' xs
	product1' _			= []

product2 :: (Num a) => [[a]] -> [[a]]
product2 = product1 . transpose

product3 :: (Num a) => [[a]] -> [[a]]
product3 = nDrop 0 . transpose . product1 . transpose . nShift 0
	where
	nShift n (x:xs) = foldl (flip ($)) x (replicate n (0:)):nShift (n+1) xs
	nShift _ []	= []
	nDrop n (x:xs)	= drop n x:nDrop (n+1) xs
	nDrop _ []	= []

product4 :: (Num a) => [[a]] -> [[a]]
product4 = nDrop 0 . transpose . product1 . transpose . nShift 0 . reverse
	where
	nShift n (x:xs) = foldl (flip ($)) x (replicate n (0:)):nShift (n+1) xs
	nShift _ []	= []
	nDrop n (x:xs)	= drop n x:nDrop (n+1) xs
	nDrop _ []	= []

product1が横、product2が縦、product3,4が斜め。と。
んー。今思えばnShiftとかはzipを使えばええんちゃうかと思い始めた。
あとで書いてみよ。

$ time ./11
70600674

real	0m0.161s
user	0m0.006s
sys	0m0.005s

Project Euler 12


三角数の数列は自然数の和で表わされ、7番目の三角数は 1 + 2 + 3 + 4 + 5 + 6 + 7 = 28 である。三角数の最初の10項は
1, 3, 6, 10, 15, 21, 28, 36, 45, 55, ...

となる。

最初の7項について、その約数を列挙すると、以下のとおり。

1: 1

3: 1,3

6: 1,2,3,6

10: 1,2,5,10

15: 1,3,5,15

21: 1,3,7,21

28: 1,2,4,7,14,28

これから、7番目の三角数である28は5つ以上の約数をもつことが分る。

では、501 個以上の約数をもつ最初の三角数はいくらか。

三角数だー。
数学的に考えるとなんか公式使えるんかなぁ。とか思いつつ強行突破。

import Lib
import List

main :: IO ()
main = print $ head $ dropWhile ((<=500).length.divisors) (triangular::[Integer])

triangular :: (Num a) => [a]
triangular = unfoldr (\(x,n) -> Just( x+n, (x+n,n+1))) (0,1)

定義まんまっす。
まんまとかいいながら最初なぜかdivisorsじゃなくてproperDivisorsを何故かmapしてて、なぜかheadしたあとにlastしてて...なにがなんだか、なコードだったのは秘密。
これを書いていたとき俺の頭には何が沸いたのだろうか？
三角数ぐらいは原理主義的に(?)やらずに公式使えばよかったかなぁ。まぁいいか。

$ time ./12
76576500

real	0m6.231s
user	0m6.092s
sys	0m0.062s

Project Euler 13


以下の50桁の数字100個の総和の上位10桁を求めよ。

37107287533902102798797998220837590246510135740250
46376937677490009712648124896970078050417018260538
74324986199524741059474233309513058123726617309629
91942213363574161572522430563301811072406154908250
23067588207539346171171980310421047513778063246676
89261670696623633820136378418383684178734361726757
28112879812849979408065481931592621691275889832738
44274228917432520321923589422876796487670272189318
47451445736001306439091167216856844588711603153276
70386486105843025439939619828917593665686757934951
62176457141856560629502157223196586755079324193331
64906352462741904929101432445813822663347944758178
92575867718337217661963751590579239728245598838407
58203565325359399008402633568948830189458628227828
80181199384826282014278194139940567587151170094390
35398664372827112653829987240784473053190104293586
86515506006295864861532075273371959191420517255829
71693888707715466499115593487603532921714970056938
54370070576826684624621495650076471787294438377604
53282654108756828443191190634694037855217779295145
36123272525000296071075082563815656710885258350721
45876576172410976447339110607218265236877223636045
17423706905851860660448207621209813287860733969412
81142660418086830619328460811191061556940512689692
51934325451728388641918047049293215058642563049483
62467221648435076201727918039944693004732956340691
15732444386908125794514089057706229429197107928209
55037687525678773091862540744969844508330393682126
18336384825330154686196124348767681297534375946515
80386287592878490201521685554828717201219257766954
78182833757993103614740356856449095527097864797581
16726320100436897842553539920931837441497806860984
48403098129077791799088218795327364475675590848030
87086987551392711854517078544161852424320693150332
59959406895756536782107074926966537676326235447210
69793950679652694742597709739166693763042633987085
41052684708299085211399427365734116182760315001271
65378607361501080857009149939512557028198746004375
35829035317434717326932123578154982629742552737307
94953759765105305946966067683156574377167401875275
88902802571733229619176668713819931811048770190271
25267680276078003013678680992525463401061632866526
36270218540497705585629946580636237993140746255962
24074486908231174977792365466257246923322810917141
91430288197103288597806669760892938638285025333403
34413065578016127815921815005561868836468420090470
23053081172816430487623791969842487255036638784583
11487696932154902810424020138335124462181441773470
63783299490636259666498587618221225225512486764533
67720186971698544312419572409913959008952310058822
95548255300263520781532296796249481641953868218774
76085327132285723110424803456124867697064507995236
37774242535411291684276865538926205024910326572967
23701913275725675285653248258265463092207058596522
29798860272258331913126375147341994889534765745501
18495701454879288984856827726077713721403798879715
38298203783031473527721580348144513491373226651381
34829543829199918180278916522431027392251122869539
40957953066405232632538044100059654939159879593635
29746152185502371307642255121183693803580388584903
41698116222072977186158236678424689157993532961922
62467957194401269043877107275048102390895523597457
23189706772547915061505504953922979530901129967519
86188088225875314529584099251203829009407770775672
11306739708304724483816533873502340845647058077308
82959174767140363198008187129011875491310547126581
97623331044818386269515456334926366572897563400500
42846280183517070527831839425882145521227251250327
55121603546981200581762165212827652751691296897789
32238195734329339946437501907836945765883352399886
75506164965184775180738168837861091527357929701337
62177842752192623401942399639168044983993173312731
32924185707147349566916674687634660915035914677504
99518671430235219628894890102423325116913619626622
73267460800591547471830798392868535206946944540724
76841822524674417161514036427982273348055556214818
97142617910342598647204516893989422179826088076852
87783646182799346313767754307809363333018982642090
10848802521674670883215120185883543223812876952786
71329612474782464538636993009049310363619763878039
62184073572399794223406235393808339651327408011116
66627891981488087797941876876144230030984490851411
60661826293682836764744779239180335110989069790714
85786944089552990653640447425576083659976645795096
66024396409905389607120198219976047599490197230297
64913982680032973156037120041377903785566085089252
16730939319872750275468906903707539413042652315011
94809377245048795150954100921645863754710598436791
78639167021187492431995700641917969777599028300699
15368713711936614952811305876380278410754449733078
40789923115535562561142322423255033685442488917353
44889911501440648020369068063960672322193204149535
41503128880339536053299340368006977710650566631954
81234880673210146739058568557934581403627822703280
82616570773948327592232845941706525094512325230608
22918802058777319719839450180888072429661980811197
77158542502016545090413245809786882778948721859617
72107838435069186155435662884062257473692284509516
20849603980134001723930671666823555245252804609722
53503534226472524250874054075591789781264330331690

なんじゃそりゃ。
まぁ、愚直に行きますけどね(・x)b

readNums :: IO [Integer]
readNums = readFile "13.txt" >>= return . map read . lines

main :: IO ()
main = readNums >>= putStrLn . take 10 . show . sum

Viva 多倍長整数。Viva Show。

$ time ./13
5537376230

real	0m0.068s
user	0m0.005s
sys	0m0.004s

Project Euler 14


正の整数に以下の式で繰り返し生成する数列を定義する。

n → n/2 (n が偶数)

n → 3n + 1 (n が奇数)

13からはじめるとこの数列は以下のようになる。
13 → 40 → 20 → 10 → 5 → 16 → 8 → 4 → 2 → 1

13から1まで10個の項になる。この数列はどのような数字からはじめても最終的には 1 になると考えられているが、まだそのことは証明されていない(コラッツ問題)

さて、100万未満の数字の中でどの数字からはじめれば一番長い数列を生成するか。

注意: 数列の途中で100万以上になってもよい

コラッツ問題だー。
まぁ、こういう問題はThe力押し！

import Lib
import Data.Function
import Data.List

collatz :: (Integral a) => a -> a
collatz n
	| n == 1	= -1
	| even n	= n `div` 2
	| otherwise	= 3 * n + 1

collatzList :: (Integral a) => a -> [a]
collatzList x = takeWhile (>0) $ iterate collatz x

main :: IO ()
main = print $ maximumBy (compare `on` (length . collatzList)) [1..(1000000::Integer)-1]

と、まぁ、愚直にGoしたわけですが...

$ time ./14
837799

real	2m13.525s
user	2m7.064s
sys	0m1.269s

遅っ！！！！！
ぬー。コラッツ問題はメモ化すると一発なきがするなぁ...。
そうでもないか。:wq