CCF 2018-3-1 跳一跳

#include<iostream>
using namespace std;
int main()
{
    char a[30];
    int sum=0;
    for(int i=0;i<30;i++)
    {
        int k;
        cin>>k;
        if(k==0)break;
        if(i==0)
        {
            if(k==1)
            {
                sum=sum+1;
                a[0]=1;
            }
            else//为k等于2的情况
            {
                sum=sum+2;
                a[0]=2;
            }
        }
        else
        {
            if(k==2)
            {
                if(a[i-1]==1)a[i]=2;
                else a[i]=a[i-1]+2;
                sum=sum+a[i];
            }
            else//当k为1时
            {
                sum=sum+1;
                a[i]=1;
            }
        }
    }
    cout<<sum;
    return 0;
}