add day-02 + benchmarks

day-02/internal/one/benchmark

@@ -0,0 +1,6 @@
+goos: linux
+goarch: amd64
+pkg: github.com/onyx-and-iris/aoc2024/day-02/internal/one
+cpu: Intel(R) Core(TM) i7-8700K CPU @ 3.70GHz
+BenchmarkMain-12    	1000000000	         0.0003352 ns/op
+ok  	github.com/onyx-and-iris/aoc2024/day-02/internal/one	0.009s

day-02/internal/one/check.go

@@ -0,0 +1,16 @@
+package one
+
+import "math"
+
+func checkWithComparator(nums []int, comparator comparator) bool {
+	for i := 1; i < len(nums); i++ {
+		if comparator(nums, i) || !isSafe(nums[i-1], nums[i]) {
+			return false
+		}
+	}
+	return true
+}
+
+func isSafe(m, n int) bool {
+	return int(math.Abs(float64(m-n))) <= 3
+}

day-02/internal/one/check_internal_test.go

@@ -0,0 +1,69 @@
+package one
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestCheckIncreased(t *testing.T) {
+	tt := map[string]struct {
+		input []int
+		want  bool
+	}{
+		"nominal": {
+			input: []int{1, 3, 6, 7, 9},
+			want:  true,
+		},
+		"unsafe": {
+			input: []int{1, 3, 2, 4, 8},
+			want:  false,
+		},
+		"two numbers the same": {
+			input: []int{1, 3, 3, 4, 8},
+			want:  false,
+		},
+	}
+
+	cmp := func(nums []int, i int) bool {
+		return nums[i-1] >= nums[i]
+	}
+
+	for name, tc := range tt {
+		t.Run(name, func(t *testing.T) {
+			got := checkWithComparator(tc.input, cmp)
+			assert.Equal(t, tc.want, got)
+		})
+	}
+}
+
+func TestCheckDecreased(t *testing.T) {
+	tt := map[string]struct {
+		input []int
+		want  bool
+	}{
+		"nominal": {
+			input: []int{7, 6, 4, 2, 1},
+			want:  true,
+		},
+		"unsafe": {
+			input: []int{9, 7, 6, 2, 1},
+			want:  false,
+		},
+		"two numbers the same": {
+			input: []int{8, 6, 4, 4, 1},
+			want:  false,
+		},
+	}
+
+	cmp := func(nums []int, i int) bool {
+		return nums[i-1] <= nums[i]
+	}
+
+	for name, tc := range tt {
+		t.Run(name, func(t *testing.T) {
+			got := checkWithComparator(tc.input, cmp)
+			assert.Equal(t, tc.want, got)
+		})
+	}
+}

day-02/internal/one/solve.go

@@ -0,0 +1,40 @@
+package one
+
+import (
+	"bufio"
+	"bytes"
+	"io"
+
+	"github.com/onyx-and-iris/aoc2024/day-02/internal/util"
+)
+
+type comparator func([]int, int) bool
+
+func Solve(data []byte) (int, error) {
+	r := bytes.NewReader(data)
+	sum := parseLines(r)
+
+	return sum, nil
+}
+
+func parseLines(r io.Reader) int {
+	var sum int
+
+	scanner := bufio.NewScanner(r)
+	for scanner.Scan() {
+		nums := util.IntsFromString(scanner.Text())
+		if check(checkWithComparator, nums, func(nums []int, i int) bool {
+			return nums[i-1] >= nums[i]
+		}) || check(checkWithComparator, nums, func(nums []int, i int) bool {
+			return nums[i-1] <= nums[i]
+		}) {
+			sum++
+		}
+	}
+
+	return sum
+}
+
+func check(fn func([]int, comparator) bool, nums []int, comparator comparator) bool {
+	return fn(nums, comparator)
+}

day-02/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 BenchmarkMain(b *testing.B) {
+	os.Stdout, _ = os.Open(os.DevNull)
+	Solve(data)
+}