adventofcode-2021/day02.py

13 lines
369 B
Python
Raw Normal View History

2021-12-03 22:43:40 +00:00
pos = [0, 0]
direction_key = {"forward": (1, 0), "down": (0, 1), "up": (0, -1)}
with open("day02.in") as f:
# fa = [i for i in f]
for x in f:
cmd = x.split()[0]
val = int(x.split()[1])
pos[0] += direction_key[cmd][0] * val
pos[1] += direction_key[cmd][1] * val
print("Current position:", pos)
print("Product:", pos[0] * pos[1])