aoc2021/d01/d01_2.py

11 lines
306 B
Python

import sys
L = list(map(int,sys.stdin.read().splitlines()))
print(sum([0,1][a<d] for (a,d) in zip(L, L[3:])))
# window1 = [A,B,C] ; window2 = [B,C,D]
# sum(Window1) <? sum(window2)
# (A+B+C) <? (B+C+D)
# (A+B+C)-B-C <? (B+C+D)-B-C
# A <? D
# eg: comparing A+B+C and B+C+D is the same as comparing A and D