CCF 201803-1 跳一跳

C++11 代碼：

#include<iostream>
using namespace std;

int main()
{
    int n, flag = 1;
    int score = 0;
    while(1) {
        cin >> n;
        if (n == 0) break;
        if (n == 1) {
            score += 1;
            flag = 1;
        }
        if (n == 2) {
            score += flag * 2;
            ++flag;
        }
    }
    cout << score;
    return 0;
}