CCF 201803-1 跳一跳(C++)

也不知道怎么想的，最初想用数组来放各个输入的值，但后来发现，每次输入之后判断条件加值就可以了，题目比较简单。

#include<iostream>
using namespace std;
int n;
int main()
{
    int sum=0;
    int num=0;
    while(cin>>n)
    {
        if(n==0)
        break;
        else if(n==1)
        {
            num=0;
            sum++;
        }
        else if(n==2)
        {
            num++;
            sum+=num*2; 
         } 
    }
    cout<<sum<<endl;
    return 0;
 }