diff options
| author | Botond Hende <nettingman@gmail.com> | 2024-12-03 21:21:47 +0100 |
|---|---|---|
| committer | Botond Hende <nettingman@gmail.com> | 2024-12-03 21:21:47 +0100 |
| commit | 755d3322abe7d94ae87e5624bb1aed85bfb30bc3 (patch) | |
| tree | ea8a7afe1071167a17e3d63f574dad662abcedf1 /2024/day2/solve2.py | |
| parent | 873d769509c56ddbe84616d9411efc51ef0b4fd3 (diff) | |
2024 day2-3, 2023 day3 2nd
Diffstat (limited to '2024/day2/solve2.py')
| -rw-r--r-- | 2024/day2/solve2.py | 42 |
1 files changed, 42 insertions, 0 deletions
diff --git a/2024/day2/solve2.py b/2024/day2/solve2.py new file mode 100644 index 0000000..8e2f8e7 --- /dev/null +++ b/2024/day2/solve2.py @@ -0,0 +1,42 @@ +from typing import Generator
+
+safe = unsafe = 0
+
+
+def is_safe_with_dampening(report: str) -> bool:
+ numbers = [int(elem) for elem in report.split()]
+ for ii in range(len(numbers)):
+ if is_safe((elem for idx, elem in enumerate(numbers) if idx != ii)):
+ return True
+
+ return False
+
+
+def is_safe(numbers: Generator[int, None, None]) -> bool:
+ last_num = 0
+ increase = True
+ for ii, num in enumerate(numbers):
+ if ii != 0:
+ if ii == 1:
+ increase = num > last_num
+
+ if (num > last_num) != increase:
+ return False
+
+ diff = abs(num - last_num)
+ if diff < 1 or diff > 3:
+ return False
+
+ last_num = num
+
+ return True
+
+
+with open("input") as f:
+ for report in f.readlines():
+ if is_safe_with_dampening(report):
+ safe += 1
+ else:
+ unsafe += 1
+
+print(f"safe: {safe}\nunsafe: {unsafe}")
\ No newline at end of file |