CCF 201803-1 跳一跳 C++

//因为以0结束，输入不为0时便一直输入。用count记录连续2的个数，用sum记录和。判断是1还是2，如果是1直接加，如果是2根据前一个输入及连续2的个数决定加几。

#include <iostream>
using namespace std;
int main()
{
	int a=3,count=0;
	int sum=0;
	while(a!=0)
	{
		cin >> a;
		if(a==2)
		{
			count++;
			sum+=2*count;
		}
		else if(a==1)
		{
			count=0;
			sum+=1;
		}
	}
	cout << sum << endl;
	return 0;
}