adventofcode-2021/02/solve02.py

18 lines
441 B
Python
Raw Normal View History

2021-12-02 06:20:43 +01:00
#!/usr/bin/env python
# forward, depth, aim
pos=[0,0,0]
with open("input01.txt", "r") as f:
for line in f:
tokens = line.split(" ", 2)
tokens[1] = int(tokens[1])
if tokens[0] == 'forward':
pos[0] += tokens[1]
pos[1] += (tokens[1]*pos[2])
if tokens[0] == 'up':
pos[2] -= tokens[1]
if tokens[0] == 'down':
pos[2] += tokens[1]
print(pos[0] * pos[1])