CSP 201803-1 跳一跳

CSP 201803-1 跳一跳

思路: 没有跳到方块的中心则 +1分,跳到方块中心 +2 *(连续跳到中心的次数)分
代码:

#include<bits/stdc++.h>
#define repn(n) for(int i=1;i<=n;i++)
#define rep(n) for(int i=0;i<n;i++)
#define pb push_back
#define ll long long
using namespace std;

int main()
{
    int x,ans,cnt;
    ans=cnt=0;
    while(1)
    {
        cin>>x;
        if(x==0)
        {
            break;
        }
        else
        {
            if(x==1)
            {
                ans+=1;
                cnt=0;
            }
            else
            {
                ans+=(2*(cnt+1));
                cnt++;
            }
        }
    }
    cout<<ans<<endl;
}