CCF-跳一跳-python3

'''
0 没跳上
1 没有跳上中心    ---获得1分
2 跳上中心      ---上一次的得分为1分或是第一次跳跃则得分为2，否则此次得分比上一次得分多两分
计算得分
'''

finput = input().split()
jump = list(map(int, finput))
score = 0
count = 0
for i in range(len(jump)):
    if jump[i] == 2:
        count += 1  # 得2分的次数
        if i == 0:  # 第一次跳跃则得2分
            score += 2
        elif jump[i - 1] == 1:  # 上一次的得分为1分,得2分
            score += 2
        else:
            score += 2*count
    else:
        count = 0
        if jump[i] == 0:  # 没跳上，结束
            break
        elif jump[i] == 1:  # 得1分
            score += 1
print(score)