C++ 代码

#include<iostream>
#include<algorithm>
#include<vector>
using namespace std;
vector<int> a;
int main()
{
    int n;
    while((scanf("%d",&n))!=EOF)
    {
        a.push_back(n);
    }
    int sum=0,tmp=0;  //tmp用于计数
    for(int i=0;i<a.size();i++)
    {
        if(a[i]==1) {
            tmp = 0;
            sum += 1;
        }
        else if(a[i]==2){
            tmp++;
            sum += tmp*2;
        }
    }
    cout<<sum<<endl;
    return 0;
}