Add day03 - 2024

1 files changed, 52 insertions, 0 deletions

diff --git a/2024/day03/main.scm b/2024/day03/main.scm
new file mode 100755
index 0000000..1a8505d
--- /dev/null
+++ b/2024/day03/main.scm
@@ -0,0 +1,52 @@
+#!/usr/bin/guile -s
+!#
+
+(add-to-load-path (format #f "~a/2024" (getcwd)))
+(add-to-load-path (format #f "~a/.." (dirname (current-filename))))
+
+(use-modules
+ (aoc-utils)
+ (srfi srfi-1)
+ (srfi srfi-9)
+ (ice-9 regex))
+
+(define mul-rx (make-regexp "mul\\(([0-9]+),([0-9]+)\\)"))
+
+(define (mul str)
+  (let ((match (regexp-exec mul-rx str)))
+    (* (string->number (match:substring match 1))
+       (string->number (match:substring match 2)))))
+
+(define (not-corrupted-mults memory)
+  (map match:substring (list-matches mul-rx memory)))
+
+(define (part1 memory)
+  (apply + (map mul (not-corrupted-mults memory))))
+
+(define or-rx (make-regexp "do\\(\\)|don't\\(\\)|mul\\([0-9]+,[0-9]+\\)"))
+
+(define (part2 memory)
+  (let ((tokens (map match:substring (list-matches or-rx memory)))
+        (enabled? #t))
+    (apply + (map (lambda (match)
+                    (cond ((string= "do()" match) (begin
+                                                    (set! enabled? #t)
+                                                    0))
+                          ((string= "don't()" match) (begin
+                                                       (set! enabled? #f)
+                                                       0))
+                          (else (if enabled?
+                                    (mul match)
+                                    0))))
+                  tokens))))
+
+(define (main)
+  (let ((memory (file->str "input.txt")))
+    (display "Part1: ")
+    (display (part1 memory))
+    (newline)
+    (display "Part2: ")
+    (display (part2 memory))
+    (newline)))
+
+(main)