45 lines
890 B
Python
Executable File
45 lines
890 B
Python
Executable File
#!/usr/bin/env python3
|
|
|
|
import sys
|
|
|
|
|
|
if __name__ == '__main__':
|
|
lines = open(sys.argv[1]).readlines()
|
|
dirs = [i.strip('\n').split(" ") for i in lines]
|
|
|
|
# challenge 1
|
|
dep = 0
|
|
pos = 0
|
|
|
|
for d in dirs:
|
|
if d[0] == 'forward':
|
|
pos += int(d[1])
|
|
elif d[0] == 'up':
|
|
dep -= int(d[1])
|
|
elif d[0] == 'down':
|
|
dep += int(d[1])
|
|
else:
|
|
print(d)
|
|
|
|
res1 = str(dep * pos)
|
|
print("challenge 1:" + res1 + "\n")
|
|
|
|
# challenge 2
|
|
dep = 0
|
|
pos = 0
|
|
aim = 0
|
|
|
|
for d in dirs:
|
|
if d[0] == 'forward':
|
|
pos += int(d[1])
|
|
dep += int(d[1]) * aim
|
|
elif d[0] == 'up':
|
|
aim -= int(d[1])
|
|
elif d[0] == 'down':
|
|
aim += int(d[1])
|
|
else:
|
|
print(d)
|
|
res2 = str(dep * pos)
|
|
print("challenge 2:" + res2 + "\n")
|
|
|