30 lines
763 B
Python
30 lines
763 B
Python
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])
|
|
|
|
# Part 2
|
|
pos = [0, 0, 0]
|
|
with open("day02.in2") as f:
|
|
for x in f:
|
|
cmd = x.split()[0]
|
|
val = int(x.split()[1])
|
|
if cmd == "down":
|
|
pos[2] += val
|
|
elif cmd == "up":
|
|
pos[2] -= val
|
|
elif cmd == "forward":
|
|
pos[0] += val
|
|
pos[1] += pos[2] * val
|
|
|
|
print("Current position:", pos)
|
|
print("Product:", pos[0] * pos[1])
|