d4

Author: davidwashere

2024/day04/day04.go

@@ -0,0 +1,87 @@
+package day04
+
+import (
+	"aoc/util"
+	"strings"
+)
+
+var (
+	grid *util.InfGrid[string]
+)
+
+func load(inputFile string) {
+	data, _ := util.ReadFileToStringSlice(inputFile)
+	grid = util.NewInfGrid[string]()
+
+	for y, line := range data {
+		for x := 0; x < len(line); x++ {
+			c := string(line[x])
+			grid.Set(c, x, y)
+		}
+	}
+}
+
+func part1(inputFile string) int {
+	load(inputFile)
+
+	count := 0
+	grid.VisitAll(func(val string, x, y int, dims ...int) {
+		if val != "X" {
+			return
+		}
+
+		fs := []func(x, y int, count int, dims ...int) []string{
+			grid.GetNMany,
+			grid.GetEMany,
+			grid.GetSMany,
+			grid.GetWMany,
+			grid.GetNEMany,
+			grid.GetSEMany,
+			grid.GetSWMany,
+			grid.GetNWMany,
+		}
+
+		for _, f := range fs {
+			if strings.Join(f(x, y, 3), "") == "MAS" {
+				count++
+			}
+		}
+	})
+
+	return count
+}
+
+func part2(inputFile string) int {
+	load(inputFile)
+
+	totCount := 0
+	grid.VisitAll(func(val string, x, y int, dims ...int) {
+		if val != "A" {
+			return
+		}
+
+		count := 0
+		if grid.Get(x-1, y-1) == "M" && grid.Get(x+1, y+1) == "S" {
+			count++
+		}
+
+		if grid.Get(x-1, y-1) == "S" && grid.Get(x+1, y+1) == "M" {
+			count++
+		}
+
+		if grid.Get(x-1, y+1) == "M" && grid.Get(x+1, y-1) == "S" {
+			count++
+		}
+
+		if grid.Get(x-1, y+1) == "S" && grid.Get(x+1, y-1) == "M" {
+			count++
+		}
+
+		if count == 2 {
+			totCount++
+		}
+
+	})
+
+	return totCount
+}

2024/day04/day04_test.go

@@ -0,0 +1,42 @@
+package day04
+
+import (
+	"fmt"
+	"testing"
+)
+
+func TestP1(t *testing.T) {
+	got := part1("sample.txt")
+	fmt.Printf("Got: %v\n", got)
+	want := 18
+	if got != want {
+		t.Errorf("got %v but want %v", got, want)
+	}
+}
+
+func TestP1_Actual(t *testing.T) {
+	got := part1("input.txt")
+	fmt.Printf("Got: %v\n", got)
+	want := 2573
+	if got != want {
+		t.Errorf("got %v but want %v", got, want)
+	}
+}
+
+func TestP2(t *testing.T) {
+	got := part2("sample.txt")
+	fmt.Printf("Got: %v\n", got)
+	want := 9
+	if got != want {
+		t.Errorf("got %v but want %v", got, want)
+	}
+}
+
+func TestP2_Actual(t *testing.T) {
+	got := part2("input.txt")
+	fmt.Printf("Got: %v\n", got)
+	want := 1850
+	if got != want {
+		t.Errorf("got %v but want %v", got, want)
+	}
+}

2024/day04/input.txt

@@ -0,0 +1,10 @@
+MMMSXXMASM
+MSAMXMSMSA
+AMXSXMAAMM
+MSAMASMSMX
+XMASAMXAMM
+XXAMMXXAMA
+SMSMSASXSS
+SAXAMASAAA
+MAMMMXMMMM
+MXMXAXMASX

2024/day04/input2.txt

@@ -578,48 +578,120 @@ func (g *InfGrid[T]) GetN(x, y int, dims ...int) T {
 	return g.Get(x, y+1, dims...)
 }
 
+// GetNMany is like GetN but returns count values
+func (g *InfGrid[T]) GetNMany(x, y int, count int, dims ...int) []T {
+	r := []T{}
+	for i := 1; i <= count; i++ {
+		r = append(r, g.Get(x, y+i, dims...))
+	}
+	return r
+}
+
 // GetE will return the value east of the given coordinate (x+1), if the coordinate is outside the extents
 // of the grid returns the default value
 func (g *InfGrid[T]) GetE(x, y int, dims ...int) T {
 	return g.Get(x+1, y, dims...)
 }
 
+// GetEMany is like GetE but returns count values
+func (g *InfGrid[T]) GetEMany(x, y int, count int, dims ...int) []T {
+	r := []T{}
+	for i := 1; i <= count; i++ {
+		r = append(r, g.Get(x+i, y, dims...))
+	}
+	return r
+}
+
 // GetS will return the value south of the given coordinate (y-1), if the coordinate is outside the extents
 // of the grid returns the default value
 func (g *InfGrid[T]) GetS(x, y int, dims ...int) T {
 	return g.Get(x, y-1, dims...)
 }
 
+// GetSMany is like GetS but returns count values
+func (g *InfGrid[T]) GetSMany(x, y int, count int, dims ...int) []T {
+	r := []T{}
+	for i := 1; i <= count; i++ {
+		r = append(r, g.Get(x, y-i, dims...))
+	}
+	return r
+}
+
 // GetW will return the value west of the given coordinate (x-1), if the coordinate is outside the extents
 // of the grid returns the default value
 func (g *InfGrid[T]) GetW(x, y int, dims ...int) T {
 	return g.Get(x-1, y, dims...)
 }
 
+// GetWMany is like GetW but returns count values
+func (g *InfGrid[T]) GetWMany(x, y int, count int, dims ...int) []T {
+	r := []T{}
+	for i := 1; i <= count; i++ {
+		r = append(r, g.Get(x-i, y, dims...))
+	}
+	return r
+}
+
 // GetNE will return the value north-east of the given coordinate (x+1, y+1), if the coordinate is outside the extents
 // of the grid returns the default value
 func (g *InfGrid[T]) GetNE(x, y int, dims ...int) T {
 	return g.Get(x+1, y+1, dims...)
 }
 
+// GetNEMany is like GetNE but returns count values
+func (g *InfGrid[T]) GetNEMany(x, y int, count int, dims ...int) []T {
+	r := []T{}
+	for i := 1; i <= count; i++ {
+		r = append(r, g.Get(x+i, y+i, dims...))
+	}
+	return r
+}
+
 // GetSE will return the value south-east of the given coordinate (x+1, y-1), if the coordinate is outside the extents
 // of the grid returns the default value
 func (g *InfGrid[T]) GetSE(x, y int, dims ...int) T {
 	return g.Get(x+1, y-1, dims...)
 }
 
+// GetSEMany is like GetSE but returns count values
+func (g *InfGrid[T]) GetSEMany(x, y int, count int, dims ...int) []T {
+	r := []T{}
+	for i := 1; i <= count; i++ {
+		r = append(r, g.Get(x+i, y-i, dims...))
+	}
+	return r
+}
+
 // GetSW will return the value south-west of the given coordinate (x-1, y-1), if the coordinate is outside the extents
 // of the grid returns the default value
 func (g *InfGrid[T]) GetSW(x, y int, dims ...int) T {
 	return g.Get(x-1, y-1, dims...)
 }
 
+// GetSWMany is like GetSW but returns count values
+func (g *InfGrid[T]) GetSWMany(x, y int, count int, dims ...int) []T {
+	r := []T{}
+	for i := 1; i <= count; i++ {
+		r = append(r, g.Get(x-i, y-i, dims...))
+	}
+	return r
+}
+
 // GetNW will return the value north-west of the given coordinate (x-1, y+1), if the coordinate is outside the extents
 // of the grid returns the default value
 func (g *InfGrid[T]) GetNW(x, y int, dims ...int) T {
 	return g.Get(x-1, y+1, dims...)
 }
 
+// GetNWMany is like GetNW but returns count values
+func (g *InfGrid[T]) GetNWMany(x, y int, count int, dims ...int) []T {
+	r := []T{}
+	for i := 1; i <= count; i++ {
+		r = append(r, g.Get(x-i, y+i, dims...))
+	}
+	return r
+}
+
 // GetOrtho will return the values north, east, south, and west of the given coordinate, if a coordinate is outside
 // the extents of the grid it will be set to the default
 func (g *InfGrid[T]) GetOrtho(x, y int, dims ...int) []T {