-
Notifications
You must be signed in to change notification settings - Fork 0
/
main.go
52 lines (40 loc) · 1.09 KB
/
main.go
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
package main
import (
"fmt"
"os"
"strconv"
"github.com/jyggen/advent-of-go/internal/solver"
"github.com/jyggen/advent-of-go/internal/utils"
)
func main() {
p1, p2, err := solver.SolveFromFile(os.Stdin, SolvePart1, SolvePart2)
if err != nil {
panic(err)
}
fmt.Println(p1)
fmt.Println(p2)
}
func growth(days int, init []int) int {
state := [9]int{}
for _, v := range init {
state[v]++
}
for i := 0; i < days; i++ {
state[8], state[7], state[6], state[5], state[4], state[3], state[2], state[1], state[0] = state[0], state[8], state[7]+state[0], state[6], state[5], state[4], state[3], state[2], state[1]
}
return state[0] + state[1] + state[2] + state[3] + state[4] + state[5] + state[6] + state[7] + state[8]
}
func SolvePart1(input string) (string, error) {
intSlice, err := utils.ToIntegerSlice(input, ",")
if err != nil {
return "", err
}
return strconv.Itoa(growth(80, intSlice)), nil
}
func SolvePart2(input string) (string, error) {
intSlice, err := utils.ToIntegerSlice(input, ",")
if err != nil {
return "", err
}
return strconv.Itoa(growth(256, intSlice)), nil
}