2018-3-18CCF认证考试第一题《跳一跳》

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