46 lines
702 B
Go
46 lines
702 B
Go
package dayone
|
|
|
|
import (
|
|
"advent-of-code/internal/registry"
|
|
"os"
|
|
"strconv"
|
|
"strings"
|
|
)
|
|
|
|
func init() {
|
|
registry.Register("2018D1", ParseInput, PartOne, PartTwo)
|
|
}
|
|
|
|
func ParseInput(filepath string) []int {
|
|
content, _ := os.ReadFile(filepath)
|
|
var data []int
|
|
for line := range strings.SplitSeq(string(content), "\n") {
|
|
num, _ := strconv.Atoi(line)
|
|
data = append(data, num)
|
|
}
|
|
return data
|
|
}
|
|
|
|
func PartOne(data []int) int {
|
|
sum := 0
|
|
for _, num := range data {
|
|
sum += num
|
|
}
|
|
return sum
|
|
}
|
|
|
|
func PartTwo(data []int) int {
|
|
seen := make(map[int]bool)
|
|
sum := 0
|
|
seen[0] = true
|
|
for {
|
|
for _, num := range data {
|
|
sum += num
|
|
if seen[sum] {
|
|
return sum
|
|
}
|
|
seen[sum] = true
|
|
}
|
|
}
|
|
}
|