18 lines
441 B
Python
18 lines
441 B
Python
#!/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])
|