CSP 2018-3-1 跳一跳

#include <iostream>
using namespace std;

int type;
int grade = 0;
int  triple= 1;
int main() 
{
	while (cin >> type )
	{
		if (type == 0)
		{
			cout << grade;
		}
		else 
		{
			if(type==1)
			{
				if (grade == 0) grade += 1;
				else
				{
					grade += 1;
					triple = 1;
				}
			}
			else
			{
					grade += 2*triple;
					triple += 1;
			}
		}
	}
	return 0;
}