CCF/CSP 201803-1 跳一跳 C++

#include<iostream>
using namespace std;
int main(){
	int a[32];//定义一个整型数组a，用于存储要输入的数字，长度为32 
	int sum=0;//定义一个整型变量sum，用于存储结果 
	int count=0;//定义一个整型变量count，用于存储跳到2的次数 
	for(int i=0;i<30;i++){
		cin>>a[i];//依次输入数字，存储到数组a中 
		if(a[i]==0){//如果输入的数字为0 
			break;//停止输入 
		}
	}
	for(int i=0;i<30;i++){//遍历整个数组a 
		if(a[i]==0){//如果发现有数字为0，立即跳出循环 
			break;
		}
		else if(a[i]==1){//如果数字为1 
			sum+=1;//得分加1 
			count=0;//将跳到2的次数初始化为0 
		}
		else if(a[i]==2){//如果数字为2 
			count++;//用于记录跳到数字2次数的count加1 
			if(a[i-1]==1){//如果上次跳到1 
				sum+=2;//得分加2 
			}
			else if(i==0){//如果是第一跳就跳到2 
				sum+=2;//得分加2 
			}
			else if(a[i-1]==2){//如果上一条是2 
				sum+=2*count;//得分加2乘跳到2的次数 
			}	
		}	
	}	
	cout<<sum<<endl;	
	return 0;
}