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
65
66
67
68
69
|
module Day5B where
import System.Environment (getArgs)
data Keymap k a = Leaf k a
| Node k (Keymap k a) (Keymap k a)
deriving (Eq, Show)
type SeatingPlan = Keymap Int (Keymap Int Bool)
data Step = Bigger | Smaller
setFromSteps :: Ord k => [Step] -> a -> Keymap k a -> Keymap k a
setFromSteps [] v (Leaf k _) = Leaf k v
setFromSteps (Bigger:xs) v (Node k l r) = Node k l (setFromSteps xs v r)
setFromSteps (Smaller:xs) v (Node k l r) = Node k (setFromSteps xs v l) r
performSteps :: Ord k => [Step] -> Keymap k a -> Keymap k a
performSteps _ k@(Leaf _ _) = k
performSteps [] k = k
performSteps (Bigger:xs) (Node _ _ r) = performSteps xs r
performSteps (Smaller:xs) (Node _ l _) = performSteps xs l
fromList :: Ord k => [(k,a)] -> Keymap k a
fromList ((k, v):[]) = Leaf k v
fromList xs = Node k (fromList (take pivot xs)) (fromList (drop pivot xs))
where pivot = length xs `div` 2
(k, _) = xs!!pivot
toList :: Ord k => Keymap k a -> [(k, a)]
toList (Leaf k v) = [(k, v)]
toList (Node _ l r) = toList l ++ toList r
constructPlan :: Int -> Int -> SeatingPlan
constructPlan rows cols = fromList [(i, fromList $ zip [0..cols - 1] (replicate cols False)) | i <- [0..rows - 1]]
toSteps :: String -> [Step]
toSteps "" = []
toSteps ('B':xs) = Bigger : toSteps xs
toSteps ('R':xs) = Bigger : toSteps xs
toSteps ('F':xs) = Smaller : toSteps xs
toSteps ('L':xs) = Smaller : toSteps xs
toSeatId :: Int -> Int -> Int
toSeatId r c = (r * 8) + c
setPassOccupied :: String -> SeatingPlan -> SeatingPlan
setPassOccupied s k = setFromSteps (take 7 steps) col' k
where steps = toSteps s
(Leaf _ col) = performSteps (take 7 steps) k
col' = setFromSteps (drop 7 steps) True col
kmFilter :: Ord k => (a -> Bool) -> Keymap k a -> [(k, a)]
kmFilter f (Leaf k v) | f v = [(k, v)]
| otherwise = []
kmFilter f (Node _ l r) = kmFilter f l ++ kmFilter f r
filterUnoccupied :: SeatingPlan -> [(Int, Int)]
filterUnoccupied k = [(r,c) | (r, col) <- toList k, (c, occupied) <- toList col, not occupied]
main :: IO ()
main = do
args <- getArgs;
content <- readFile $ head args;
let l = lines content;
let p = constructPlan 128 8;
let full = foldr setPassOccupied p l;
print $ filterUnoccupied full;
|
