advent/2017/day05.py

29 lines
322 B
Python
Raw Permalink Normal View History

2017-12-14 00:48:23 +00:00
with open('data05.txt') as f:
data = f.readlines()
jumps = list(map(int, data))
steps = 0
i = 0
'''
# part 1
while i < len(jumps):
old = i
i += jumps[i]
jumps[old] += 1
steps += 1
'''
# part 2
while i < len(jumps):
j = jumps[i]
if j > 2:
jumps[i] -= 1
else:
jumps[i] += 1
i += j
steps += 1
print(steps)