chore(aoc-2024): day2 part 2

Signed-off-by: Márk Kővári <kovarimarkofficial@gmail.com>

Author: markkovari

2024/cmd/02.go

@@ -44,7 +44,7 @@ func countSafeLevels(levels [][]int) int {
 }
 
 func isLevelSafe(row []int) bool {
-	if len(row) < 3 {
+	if len(row) < 2 {
 		return false
 	}
 	increasing := row[0] < row[1]
@@ -62,3 +62,33 @@ func isLevelSafe(row []int) bool {
 	}
 	return true
 }
+
+func isLevelSafeTolerate(row []int) bool {
+	for i := 0; i < len(row); i++ {
+		if isLevelSafe(skipAtFrom(i, row)) {
+			return true
+		}
+	}
+	return false
+}
+
+func countSafeLevelsTolerate(levels [][]int) int {
+	count := 0
+	for _, row := range levels {
+		if isLevelSafeTolerate(row) {
+			count++
+		}
+	}
+	return count
+}
+
+func skipAtFrom(index int, elements []int) []int {
+	newElements := make([]int, 0)
+	for i := range len(elements) {
+		if i == index {
+			continue
+		}
+		newElements = append(newElements, elements[i])
+	}
+	return newElements
+}

2024/cmd/02_test.go

@@ -9,7 +9,7 @@ func Test02Example(t *testing.T) {
 	}
 	countSafe := countSafeLevels(levels)
 	if countSafe != 2 {
-		t.Fatalf("Expected 2 safe leves, got=%d", countSafe)
+		t.Fatalf("Expected 2 safe levels, got=%d", countSafe)
 	}
 }
 
@@ -20,6 +20,28 @@ func Test02Input(t *testing.T) {
 	}
 	countSafe := countSafeLevels(levels)
 	if countSafe != 686 {
-		t.Fatalf("Expected 686 safe leves, got=%d", countSafe)
+		t.Fatalf("Expected 686 safe levels, got=%d", countSafe)
+	}
+}
+
+func Test02Part2Example(t *testing.T) {
+	levels, err := read02File("../data/02/example")
+	if err != nil {
+		t.Fatalf("Error reading file: %v", err)
+	}
+	countSafe := countSafeLevelsTolerate(levels)
+	if countSafe != 4 {
+		t.Fatalf("Expected 4 safe levels, got=%d", countSafe)
+	}
+}
+
+func Test02Part2Input(t *testing.T) {
+	levels, err := read02File("../data/02/input")
+	if err != nil {
+		t.Fatalf("Error reading file: %v", err)
+	}
+	countSafe := countSafeLevelsTolerate(levels)
+	if countSafe != 717 {
+		t.Fatalf("Expected 717 safe levels, got=%d", countSafe)
 	}
 }