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
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
|
import Control.Monad.State
import Data.Foldable
import Data.Maybe
import Data.Sequence as Seq
import Data.Tuple
import System.Random
-- | Take a random element from a list, removing it from the list.
randomChoice :: RandomGen g => State (Seq a, g) (Maybe a)
randomChoice =
get >>=
(\(xs, g) ->
if (Seq.null xs)
then (return Nothing)
else (let (r, g2) = randomR (0, Seq.length xs - 1) g
in (put (deleteAt r xs, g2) >> (return (Just $ index xs r)))))
-- | Returns a list of randomly chosen values with no repeats.
randomChoices :: RandomGen g => Int -> State (Seq a, g) (Seq a)
randomChoices n =
state
(\si ->
let (maybeVals, sf) = runState (sequence $ Prelude.replicate n randomChoice) si
in (fromList $ catMaybes maybeVals, sf))
data Bacterium
= Red
| Green
deriving (Show)
type Bacteria = Seq Bacterium
reproduce :: Bacteria -> Bacteria
reproduce b = b <> b
eat :: RandomGen g => (Bacteria, g) -> (Bacteria, g)
eat (b, g) = snd $ runState (randomChoices $ Seq.length b `quot` 2) (b, g)
hour :: RandomGen g => (Bacteria, g) -> (Bacteria, g)
hour (b, g) = eat (reproduce b, g)
initialize :: Int -> Bacteria
initialize n = (Seq.replicate (n `quot` 2) Red) <> (Seq.replicate (n `quot` 2) Green)
isRed :: Bacterium -> Bool
isRed Red = True
isRed Green = False
nRed :: Bacteria -> Int
nRed b = Seq.length $ Seq.filter isRed b
isDone :: Bacteria -> Bool
isDone b
| nRed b == 0 = True
| nRed b == Seq.length b = True
| otherwise = False
run :: RandomGen g => Int -> g -> ([Bacteria], g)
run n g = (fst <$> steps, snd $ last steps)
where
steps = takeWhile (not . isDone . fst) $ iterate hour (initialize n, g)
lifetime :: [Bacteria] -> Int
lifetime b = Data.Foldable.length b
|