aoc2021/2/a.py

20 lines
323 B
Python

with open("input.txt") as f:
lines = [i for i in f.readlines()]
depth = 0
hpos = 0
for line in lines:
verb, val = line.strip().split()
val = int(val)
if verb == "forward":
hpos += val
elif verb == "down":
depth += val
elif verb == "up":
depth -= val
print(depth * hpos)