From c3fa65e4a85d43fee7999389a74604fe6f65c630 Mon Sep 17 00:00:00 2001 From: onyx-and-iris Date: Thu, 19 Dec 2024 09:39:38 +0000 Subject: [PATCH] add day-19 + benchmarks --- day-19/benchmark | 15 +++++ day-19/cmd/cli/main.go | 41 ++++++++++++ day-19/go.mod | 7 +++ day-19/go.sum | 15 +++++ day-19/internal/one/benchmark | 6 ++ day-19/internal/one/solve.go | 58 +++++++++++++++++ day-19/internal/one/solve_internal_test.go | 15 +++++ day-19/internal/one/util.go | 33 ++++++++++ day-19/internal/two/benchmark | 6 ++ day-19/internal/two/solve.go | 72 ++++++++++++++++++++++ day-19/internal/two/solve_internal_test.go | 15 +++++ day-19/internal/two/util.go | 33 ++++++++++ day-19/makefile | 30 +++++++++ day-19/solve.go | 20 ++++++ day-19/solve_internal_test.go | 15 +++++ 15 files changed, 381 insertions(+) create mode 100644 day-19/benchmark create mode 100644 day-19/cmd/cli/main.go create mode 100644 day-19/go.mod create mode 100644 day-19/go.sum create mode 100644 day-19/internal/one/benchmark create mode 100644 day-19/internal/one/solve.go create mode 100644 day-19/internal/one/solve_internal_test.go create mode 100644 day-19/internal/one/util.go create mode 100644 day-19/internal/two/benchmark create mode 100644 day-19/internal/two/solve.go create mode 100644 day-19/internal/two/solve_internal_test.go create mode 100644 day-19/internal/two/util.go create mode 100644 day-19/makefile create mode 100644 day-19/solve.go create mode 100644 day-19/solve_internal_test.go diff --git a/day-19/benchmark b/day-19/benchmark new file mode 100644 index 0000000..303ea75 --- /dev/null +++ b/day-19/benchmark @@ -0,0 +1,15 @@ +goos: linux +goarch: amd64 +pkg: github.com/onyx-and-iris/aoc2024/day-19 +cpu: Intel(R) Core(TM) i7-8700K CPU @ 3.70GHz +BenchmarkSolve-12 1000000000 0.1996 ns/op +BenchmarkSolve-12 1000000000 0.2115 ns/op +BenchmarkSolve-12 1000000000 0.1903 ns/op +BenchmarkSolve-12 1000000000 0.1840 ns/op +BenchmarkSolve-12 1000000000 0.1891 ns/op +BenchmarkSolve-12 1000000000 0.1828 ns/op +BenchmarkSolve-12 1000000000 0.1888 ns/op +BenchmarkSolve-12 1000000000 0.1834 ns/op +BenchmarkSolve-12 1000000000 0.1895 ns/op +BenchmarkSolve-12 1000000000 0.1832 ns/op +ok github.com/onyx-and-iris/aoc2024/day-19 24.811s diff --git a/day-19/cmd/cli/main.go b/day-19/cmd/cli/main.go new file mode 100644 index 0000000..3ea5c19 --- /dev/null +++ b/day-19/cmd/cli/main.go @@ -0,0 +1,41 @@ +/******************************************************************************** + Advent of Code 2024 - day-19 +********************************************************************************/ + +package main + +import ( + "embed" + "flag" + "fmt" + "slices" + + log "github.com/sirupsen/logrus" + + problems "github.com/onyx-and-iris/aoc2024/day-19" +) + +//go:embed testdata +var files embed.FS + +func main() { + filename := flag.String("f", "input.txt", "input file") + loglevel := flag.Int("l", int(log.InfoLevel), "log level") + flag.Parse() + + if slices.Contains(log.AllLevels, log.Level(*loglevel)) { + log.SetLevel(log.Level(*loglevel)) + } + + data, err := files.ReadFile(fmt.Sprintf("testdata/%s", *filename)) + if err != nil { + log.Fatal(err) + } + + one, two, err := problems.Solve(data) + if err != nil { + log.Fatal(err) + } + + fmt.Printf("solution one: %d\nsolution two: %d\n", one, two) +} diff --git a/day-19/go.mod b/day-19/go.mod new file mode 100644 index 0000000..37604c1 --- /dev/null +++ b/day-19/go.mod @@ -0,0 +1,7 @@ +module github.com/onyx-and-iris/aoc2024/day-19 + +go 1.23.3 + +require github.com/sirupsen/logrus v1.9.3 + +require golang.org/x/sys v0.0.0-20220715151400-c0bba94af5f8 // indirect diff --git a/day-19/go.sum b/day-19/go.sum new file mode 100644 index 0000000..21f9bfb --- /dev/null +++ b/day-19/go.sum @@ -0,0 +1,15 @@ +github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= +github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c= +github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= +github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM= +github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4= +github.com/sirupsen/logrus v1.9.3 h1:dueUQJ1C2q9oE3F7wvmSGAaVtTmUizReu6fjN8uqzbQ= +github.com/sirupsen/logrus v1.9.3/go.mod h1:naHLuLoDiP4jHNo9R0sCBMtWGeIprob74mVsIT4qYEQ= +github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME= +github.com/stretchr/testify v1.7.0 h1:nwc3DEeHmmLAfoZucVR881uASk0Mfjw8xYJ99tb5CcY= +github.com/stretchr/testify v1.7.0/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg= +golang.org/x/sys v0.0.0-20220715151400-c0bba94af5f8 h1:0A+M6Uqn+Eje4kHMK80dtF3JCXC4ykBgQG4Fe06QRhQ= +golang.org/x/sys v0.0.0-20220715151400-c0bba94af5f8/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= +gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0= +gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c h1:dUUwHk2QECo/6vqA44rthZ8ie2QXMNeKRTHCNY2nXvo= +gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM= diff --git a/day-19/internal/one/benchmark b/day-19/internal/one/benchmark new file mode 100644 index 0000000..bc2a8b5 --- /dev/null +++ b/day-19/internal/one/benchmark @@ -0,0 +1,6 @@ +goos: linux +goarch: amd64 +pkg: github.com/onyx-and-iris/aoc2024/day-19/internal/one +cpu: Intel(R) Core(TM) i7-8700K CPU @ 3.70GHz +BenchmarkSolve-12 1000000000 0.1706 ns/op +ok github.com/onyx-and-iris/aoc2024/day-19/internal/one 2.018s diff --git a/day-19/internal/one/solve.go b/day-19/internal/one/solve.go new file mode 100644 index 0000000..bc21db8 --- /dev/null +++ b/day-19/internal/one/solve.go @@ -0,0 +1,58 @@ +package one + +import ( + "bytes" + "cmp" + "slices" + "strings" + + log "github.com/sirupsen/logrus" +) + +func Solve(buf []byte) (int, error) { + r := bytes.NewReader(buf) + stripes, designs, err := parseLines(r) + if err != nil { + return 0, err + } + + log.Debug(stripes, designs) + + var sum int + for _, design := range designs { + ok := isPossibleArrangement(design, stripes) + if ok { + log.Debugf("design: %s is possible", design) + sum++ + } + } + + return sum, nil +} + +func isPossibleArrangement(design string, stripes []string) bool { + if len(design) == 0 { + return true + } + + greedy := []string{} + for _, stripe := range stripes { + log.Debugf("testing design: %s with stripe: %s", design, stripe) + if strings.HasPrefix(design, stripe) { + log.Debugf("design: %s has prefix %s", design, stripe) + + greedy = append(greedy, stripe) + } + } + + slices.SortFunc(greedy, func(a, b string) int { + return cmp.Compare(len(b), len(a)) + }) + + for _, g := range greedy { + if isPossibleArrangement(design[len(g):], stripes) { + return true + } + } + return false +} diff --git a/day-19/internal/one/solve_internal_test.go b/day-19/internal/one/solve_internal_test.go new file mode 100644 index 0000000..e5fd7ea --- /dev/null +++ b/day-19/internal/one/solve_internal_test.go @@ -0,0 +1,15 @@ +package one + +import ( + _ "embed" + "os" + "testing" +) + +//go:embed testdata/input.txt +var data []byte + +func BenchmarkSolve(b *testing.B) { + os.Stdout, _ = os.Open(os.DevNull) + Solve(data) +} diff --git a/day-19/internal/one/util.go b/day-19/internal/one/util.go new file mode 100644 index 0000000..e87214c --- /dev/null +++ b/day-19/internal/one/util.go @@ -0,0 +1,33 @@ +package one + +import ( + "bufio" + "io" + "strings" +) + +func parseLines(r io.Reader) ([]string, []string, error) { + var stripes, designs []string + + var inDesigns bool + scanner := bufio.NewScanner(r) + for scanner.Scan() { + line := scanner.Text() + if line == "" { + inDesigns = true + continue + } + + if inDesigns { + designs = append(designs, line) + } else { + stripes = strings.Split(line, ", ") + } + } + + if err := scanner.Err(); err != nil { + return nil, nil, err + } + + return stripes, designs, nil +} diff --git a/day-19/internal/two/benchmark b/day-19/internal/two/benchmark new file mode 100644 index 0000000..911c93d --- /dev/null +++ b/day-19/internal/two/benchmark @@ -0,0 +1,6 @@ +goos: linux +goarch: amd64 +pkg: github.com/onyx-and-iris/aoc2024/day-19/internal/two +cpu: Intel(R) Core(TM) i7-8700K CPU @ 3.70GHz +BenchmarkSolve-12 1000000000 0.01208 ns/op +ok github.com/onyx-and-iris/aoc2024/day-19/internal/two 0.091s diff --git a/day-19/internal/two/solve.go b/day-19/internal/two/solve.go new file mode 100644 index 0000000..2f2a5e3 --- /dev/null +++ b/day-19/internal/two/solve.go @@ -0,0 +1,72 @@ +package two + +import ( + "bytes" + "cmp" + "slices" + "strings" + + log "github.com/sirupsen/logrus" +) + +func Solve(buf []byte) (int, error) { + r := bytes.NewReader(buf) + stripes, designs, err := parseLines(r) + if err != nil { + return 0, err + } + + conc := len(designs) + countChan := make(chan int) + for _, design := range designs { + go func() { + count, ok := numPossibleArrangement(design, stripes, make(map[string]int)) + if ok { + log.Debugf("design: %s is possible in %d ways", design, count) + countChan <- count + return + } + countChan <- 0 + }() + } + + var totalCount int + for range conc { + count := <-countChan + totalCount += count + } + + return totalCount, nil +} + +func numPossibleArrangement(design string, stripes []string, memo map[string]int) (int, bool) { + if len(design) == 0 { + return 1, true + } + + greedy := []string{} + for _, stripe := range stripes { + if strings.HasPrefix(design, stripe) { + greedy = append(greedy, stripe) + } + } + + slices.SortFunc(greedy, func(a, b string) int { + return cmp.Compare(len(b), len(a)) + }) + + var totalCount int + for _, g := range greedy { + if count, ok := memo[design[len(g):]]; ok { + totalCount += count + continue + } + + count, ok := numPossibleArrangement(design[len(g):], stripes, memo) + if ok { + totalCount += count + } + memo[design[len(g):]] = count + } + return totalCount, true +} diff --git a/day-19/internal/two/solve_internal_test.go b/day-19/internal/two/solve_internal_test.go new file mode 100644 index 0000000..ecb05d5 --- /dev/null +++ b/day-19/internal/two/solve_internal_test.go @@ -0,0 +1,15 @@ +package two + +import ( + _ "embed" + "os" + "testing" +) + +//go:embed testdata/input.txt +var data []byte + +func BenchmarkSolve(b *testing.B) { + os.Stdout, _ = os.Open(os.DevNull) + Solve(data) +} diff --git a/day-19/internal/two/util.go b/day-19/internal/two/util.go new file mode 100644 index 0000000..f96b9ab --- /dev/null +++ b/day-19/internal/two/util.go @@ -0,0 +1,33 @@ +package two + +import ( + "bufio" + "io" + "strings" +) + +func parseLines(r io.Reader) ([]string, []string, error) { + var stripes, designs []string + + var inDesigns bool + scanner := bufio.NewScanner(r) + for scanner.Scan() { + line := scanner.Text() + if line == "" { + inDesigns = true + continue + } + + if inDesigns { + designs = append(designs, line) + } else { + stripes = strings.Split(line, ", ") + } + } + + if err := scanner.Err(); err != nil { + return nil, nil, err + } + + return stripes, designs, nil +} diff --git a/day-19/makefile b/day-19/makefile new file mode 100644 index 0000000..ea3d0ec --- /dev/null +++ b/day-19/makefile @@ -0,0 +1,30 @@ +program = day-19 + +GO = go +SRC_DIR := src +BIN_DIR := bin + +EXE := $(BIN_DIR)/$(program) + +.DEFAULT_GOAL := build + +.PHONY: fmt vet build bench clean +fmt: + $(GO) fmt ./... + +vet: fmt + $(GO) vet ./... + +build: vet | $(BIN_DIR) + $(GO) build -o $(EXE) ./$(SRC_DIR) + +bench: + $(GO) test ./internal/one/ -bench=. > internal/one/benchmark + $(GO) test ./internal/two/ -bench=. > internal/two/benchmark + $(GO) test . -count=10 -bench=. > benchmark + +$(BIN_DIR): + @mkdir -p $@ + +clean: + @rm -rv $(BIN_DIR) diff --git a/day-19/solve.go b/day-19/solve.go new file mode 100644 index 0000000..53eebdd --- /dev/null +++ b/day-19/solve.go @@ -0,0 +1,20 @@ +package dayeighteen + +import ( + "github.com/onyx-and-iris/aoc2024/day-19/internal/one" + "github.com/onyx-and-iris/aoc2024/day-19/internal/two" +) + +func Solve(buf []byte) (int, int, error) { + answerOne, err := one.Solve(buf) + if err != nil { + return 0, 0, err + } + + answerTwo, err := two.Solve(buf) + if err != nil { + return 0, 0, err + } + + return answerOne, answerTwo, nil +} diff --git a/day-19/solve_internal_test.go b/day-19/solve_internal_test.go new file mode 100644 index 0000000..c6350ad --- /dev/null +++ b/day-19/solve_internal_test.go @@ -0,0 +1,15 @@ +package dayeighteen + +import ( + _ "embed" + "os" + "testing" +) + +//go:embed testdata/input.txt +var data []byte + +func BenchmarkSolve(b *testing.B) { + os.Stdout, _ = os.Open(os.DevNull) + Solve(data) +}