CCF201803-1 跳一跳

#include<iostream>
#include<cstdio>
using namespace std;

int main(){
    int score = 0;
    int finalscore = 0;
    int a;
    while(scanf("%d",&a)!=EOF&&a){
        finalscore += a;
        if(a==1){
            score = 0;
        }
        else if(a==2){
            score += 2;
            finalscore += (score-2);
        }
    }
    printf("%d\n",finalscore);
    return 0;
}