Files
advent_of_code/2021/01.py
2021-12-02 10:36:43 -05:00

27 lines
582 B
Python

with open('01.txt', 'r') as input_file:
input_list = input_file.readlines()
# A
for i in range(0, len(input_list)):
input_list[i] = int(input_list[i].replace(r'\n', ''))
greater_counter = 0
for i in range(1, len(input_list)):
if input_list[i] > input_list[i - 1]:
greater_counter += 1
print(greater_counter)
# B
greater_counter = 0
for i in range(3, len(input_list)):
# Get sum of current 3
current = sum(input_list[i-3:i])
previous = sum(input_list[i-4:i-1])
if current > previous:
greater_counter += 1
print(greater_counter)