/ Python / 2017 / 05.py
05.py
 1  from lib import *
 2  
 3  input = read_input(2017, 5)
 4  
 5  lines = input.splitlines()
 6  
 7  
 8  jumps = [*map(int, lines)]
 9  pos = 0
10  i = 0
11  while pos in range(len(jumps)):
12      p = pos
13      pos += jumps[pos]
14      jumps[p] += 1
15      i += 1
16  
17  print(i)
18  
19  
20  jumps = [*map(int, lines)]
21  pos = 0
22  i = 0
23  while pos in range(len(jumps)):
24      p = pos
25      pos += jumps[pos]
26      if jumps[p] >= 3:
27          jumps[p] -= 1
28      else:
29          jumps[p] += 1
30      i += 1
31  
32  print(i)