feat: add day 1 2024

aoc2023/day14/main.go

@@ -4,6 +4,7 @@ import (
 	"bufio"
 	_ "embed"
 	"fmt"
+	"slices"
 	"strings"
 
 	aoc "go.sour.is/advent-of-code"
@@ -62,7 +63,7 @@ func run(scan *bufio.Scanner) (*result, error) {
 					func(i int, v int, counts [3]int) [3]int {
 						counts[v]++
 						return counts
-					}, [3]int{}, maps.Values(memo)...)
+					}, [3]int{}, slices.Values(maps.Values(memo)))
 
 				// fmt.Println(i, counts)
 				i = 1_000_000_000 - (1_000_000_000-counts[0]-counts[1])%counts[2]

aoc2023/day15/main.go

@@ -32,14 +32,14 @@ func run(scan *bufio.Scanner) (*result, error) {
 		r.valuePT1 = aoc.Reduce(func(i int, t string, sum int) int {
 			sum += hash(t)
 			return sum
-		}, 0, ops...)
+		}, 0, slices.Values(ops))
 	}
 
 	var boxen boxes
 
 	boxen = aoc.Reduce(func(i int, op string, b boxes) boxes {
 		return b.Op(op)
-	}, boxen, ops...)
+	}, boxen, slices.Values(ops))
 
 	r.valuePT2 = boxen.Sum()
 

aoc2024/day01/example.txt

@@ -0,0 +1,6 @@
+3   4
+4   3
+2   5
+1   3
+3   9
+3   3

aoc2024/day01/main.go

@@ -4,6 +4,9 @@ import (
 	"bufio"
 	_ "embed"
 	"fmt"
+	"iter"
+	"slices"
+	"sort"
 
 	aoc "go.sour.is/advent-of-code"
 )
@@ -20,11 +23,64 @@ type result struct {
 func (r result) String() string { return fmt.Sprintf("%#v", r) }
 
 func run(scan *bufio.Scanner) (*result, error) {
+	var (
+		left  []int
+		right []int
+	)
 
 	for scan.Scan() {
-		_ = scan.Text()
+		txt := scan.Text()
 
+		var l, r int
+		_, err := fmt.Sscanf(txt, "%d %d", &l, &r)
+		if err != nil {
+			return nil, err
 		}
 
-	return &result{}, nil
+		left = append(left, l)
+		right = append(right, r)
 	}
+
+	sort.Ints(left)
+	sort.Ints(right)
+
+	result := &result{}
+
+	result.valuePT1 = aoc.Reduce(func(i int, z pair[int, int], sum int) int {
+		return sum + aoc.ABS(z.L-z.R)
+	}, 0, zip(slices.Values(left), slices.Values(right)))
+
+	rmap := aoc.Reduce(func(i int, z int, m map[int]int) map[int]int {
+		m[z]++
+		return m
+	}, make(map[int]int), slices.Values(right))
+	
+	for _, v := range left {
+		if r, ok := rmap[v]; ok {	
+			result.valuePT2 += v*r
+		}
+	}
+
+	return result, nil
+}
+
+type pair[L, R any] struct {
+	L L
+	R R
+}
+
+func zip[L, R any](l iter.Seq[L], r iter.Seq[R]) iter.Seq[pair[L,R]] {
+	return func(yield func(pair[L, R]) bool) {
+		pullR, stop := iter.Pull(r)
+		defer stop()
+
+		for l := range l {
+			r, _ := pullR()
+
+			if !yield(pair[L, R]{L: l, R: r}) {
+				return
+			}
+		}
+	}
+}
+

aoc2024/day01/main_test.go

@@ -24,8 +24,8 @@ func TestExample(t *testing.T) {
 	is.NoErr(err)
 
 	t.Log(result)
-	is.Equal(result.valuePT1, 0)
+	is.Equal(result.valuePT1, 11)
-	is.Equal(result.valuePT2, 0)
+	is.Equal(result.valuePT2, 31)
 }
 
 func TestSolution(t *testing.T) {
@@ -36,6 +36,6 @@ func TestSolution(t *testing.T) {
 	is.NoErr(err)
 
 	t.Log(result)
-	is.Equal(result.valuePT1, 0)
+	is.Equal(result.valuePT1, 2756096)
-	is.Equal(result.valuePT2, 0)
+	is.Equal(result.valuePT2, 23117829)
 }

itertools.go

@@ -1,6 +1,7 @@
 package aoc
 
 import (
+	"iter"
 	"strconv"
 )
 
@@ -17,8 +18,8 @@ func Repeat[T any](s T, i int) []T {
 	return lis
 }
 
-func Reduce[T, U any](fn func(int, T, U) U, u U, list ...T) U {
+func Reduce[T, U any](fn func(int, T, U) U, u U, list iter.Seq[T]) U {
-	for i, t := range list {
+	for i, t := range Enumerate(list) {
 		u = fn(i, t, u)
 	}
 	return u
@@ -54,3 +55,14 @@ func Pairwise[T any](arr []T) [][2]T {
 	}
 	return pairs
 }
+
+func Enumerate[T any](arr iter.Seq[T]) iter.Seq2[int, T] {
+	return func(yield func(int, T) bool) {
+		i := 0
+		for v := range arr {
+			if !yield(i, v) {
+				return
+			}
+		}
+	}
+}