diff --git a/5/5p2.py b/5/5p2.py index bc759ba..175319b 100755 --- a/5/5p2.py +++ b/5/5p2.py @@ -16,11 +16,14 @@ def solve(filename): while True: oldpos = pos - pos = pos + mylist[pos] - if mylist[oldpos] >= 3: - mylist[oldpos] = mylist[oldpos] - 1 + oldvalue = mylist[oldpos] + pos = pos + oldvalue + + if oldvalue >= 3: + mylist[oldpos] = oldvalue - 1 else: - mylist[oldpos] = mylist[oldpos] + 1 + mylist[oldpos] = oldvalue + 1 + steps = steps + 1 if pos > total: break