wip

2024/02.odin

@@ -1,14 +1,65 @@
 package aoc
 
 import "core:fmt"
-import "core:log"
-import "core:os"
+import "core:math"
+import "core:strconv"
+import "core:strings"
 
 day02 :: proc() {
-    input := slurp_lines("input/02")
+	input := slurp_lines("input/02")
 	defer delete(input)
-	
+
+	p1: int
+	p2: int
 	for line in input {
-	    
+		sp := strings.split(line, " ")
+		report := make([dynamic]int, 0, len(sp))
+		for s in sp {
+			i, _ := strconv.parse_int(s)
+			append(&report, i)
+		}
+		if validate(report) {
+			p1 += 1
+		}
+
+		p2_valid := true
+		for i := 0; i < len(report); i += 1 {
+			cpy := make([dynamic]int, 0, len(report))
+			copy(cpy[:], report[:])
+			ordered_remove(&cpy, i)
+			if !validate(cpy) {
+				p2_valid = false
+				delete(cpy)
+				break
+			}
+			delete(cpy)
+		}
+		if p2_valid {
+			p2 += 1
+		}
+		delete(report)
 	}
+	fmt.println(p1)
+	fmt.println(p2)
+}
+
+validate :: proc(report: [dynamic]int) -> bool {
+	increasing := false
+	if report[0] < report[1] {
+		increasing = true
+	}
+
+	for i := 1; i < len(report); i += 1 {
+		if increasing && report[i - 1] > report[i] {
+			return false
+		}
+		if !increasing && report[i - 1] < report[i] {
+			return false
+		}
+		diff := math.abs(report[i - 1] - report[i])
+		if diff < 1 || diff > 3 {
+			return false
+		}
+	}
+	return true
 }