49 lines
922 B
Python
49 lines
922 B
Python
with open('input.text', 'r') as file:
|
|
data: list[str] = file.readlines()
|
|
|
|
# Parse Input
|
|
|
|
cases = []
|
|
|
|
for line in data:
|
|
cases.append([int(s) for s in line.split(' ')])
|
|
|
|
def test_levels(levels: list[int]) -> bool:
|
|
negative = levels[0] - levels[1] <= 0
|
|
|
|
for i in range(0, len(levels) - 1):
|
|
change = levels[i] - levels[i + 1]
|
|
|
|
if (change <= 0) != negative or abs(change) > 3 or change == 0:
|
|
return False
|
|
|
|
return True
|
|
|
|
# Part 1
|
|
|
|
total = 0
|
|
for levels in cases:
|
|
if test_levels(levels):
|
|
total += 1
|
|
|
|
print(f'Part 1: {total}')
|
|
|
|
# Part 2
|
|
|
|
total = 0
|
|
for levels in cases:
|
|
safe = test_levels(levels)
|
|
|
|
if safe:
|
|
total += 1
|
|
else:
|
|
for i in range(0, len(levels)):
|
|
temp_levels = levels.copy()
|
|
temp_levels.pop(i)
|
|
|
|
if test_levels(temp_levels):
|
|
total += 1
|
|
break
|
|
|
|
print(f'Part 2: {total}')
|