CSP跳一跳python

#100
a=list(map(int,input().split()))
s=0
c=0
for i in range(len(a)):
    if a[i]==0:
        continue
    if a[i]==1:
        s+=1
    if a[i]==2:
        s+=2
        c+=1
        if c>=1 and a[i-1]==2:
            s+=c*2
        else:
            c=0
print(s)