CCF认证 2018-03 跳一跳

简单的小水题

#include
#include
#include
#include
#include
using namespace std;
typedef long long ll;
int main()
{
    int a,b=2;
    int ans=0;
    while(~scanf("%d",&a))
    {
        if(a==2)
        {
            ans+=b;
            b+=2;
        }
        else if(a==1)
        {
            ans++;
            b=2;
        }
        else break;
    }
    printf("%d\n",ans);
    return 0;
}