summaryrefslogtreecommitdiff
path: root/15_2020.py
blob: e1eac7004ab105a5bf8c9aeb8de80c5733e58f32 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
import fileinput
import re

S = [14, 8, 16, 0, 1, 17]
last_index = {}

for i, n in enumerate(S):
    if i != len(S) - 1:
        last_index[n] = i

while len(S) < 30000000:
    prev = S[-1]
    prev_prev = last_index.get(prev, -1)
    last_index[prev] = len(S) - 1
    if prev_prev == -1:
        next_ = 0
    else:
        next_ = len(S) - 1 - prev_prev
    S.append(next_)
    if len(S) == 2020:
        print(next_)
print(S[-1])