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
70
71
72
73
74
75
76
77
78
79
80
81
82
83
|
package aoc23
import (
"io"
"strings"
"testing"
)
type Game struct {
id int
cubes []Rgb
}
type Rgb struct{ r, g, b int }
func parseGame(line string) Game {
parts := strings.Split(line, ": ")
id := requireInt(parts[0][5:])
var cubes []Rgb
for _, p := range strings.Split(parts[1], "; ") {
c := Rgb{}
for _, p2 := range strings.Split(p, ", ") {
parts := strings.Split(p2, " ")
count := requireInt(parts[0])
color := parts[1]
switch color {
case "red":
c.r = count
case "blue":
c.b = count
case "green":
c.g = count
}
}
cubes = append(cubes, c)
}
return Game{id: id, cubes: cubes}
}
func validGame(game Game, rgb Rgb) bool {
return none(game.cubes, func(cubes Rgb) bool {
return cubes.r > rgb.r || cubes.g > rgb.g || cubes.b > rgb.b
})
}
func countValidGames(r io.Reader) int {
games := parseLines(r, parseGame)
valid := filter(games, partial(validGame, Rgb{12, 13, 14}))
return sum(transform(valid, func(g Game) int { return g.id }))
}
func minCubes(game Game) Rgb {
r, g, b := 0, 0, 0
for _, c := range game.cubes {
r = max(r, c.r)
g = max(g, c.g)
b = max(b, c.b)
}
return Rgb{r, g, b}
}
func minCubesProduct(r io.Reader) int {
games := parseLines(r, parseGame)
products := transform(games, compose(compose(minCubes, func(rgb Rgb) []int {
return []int{rgb.r, rgb.g, rgb.b}
}), product))
return sum(products)
}
func TestDay02(t *testing.T) {
example := `
Game 1: 3 blue, 4 red; 1 red, 2 green, 6 blue; 2 green
Game 2: 1 blue, 2 green; 3 green, 4 blue, 1 red; 1 green, 1 blue
Game 3: 8 green, 6 blue, 20 red; 5 blue, 4 red, 13 green; 5 green, 1 red
Game 4: 1 green, 3 red, 6 blue; 3 green, 6 red; 3 green, 15 blue, 14 red
Game 5: 6 red, 1 blue, 3 green; 2 blue, 1 red, 2 green
`
assert(t, 8, run(countValidGames, inputString(example)))
assert(t, 3035, run(countValidGames, inputFile(2)))
assert(t, 2286, run(minCubesProduct, inputString(example)))
assert(t, 66027, run(minCubesProduct, inputFile(2)))
}
|
