CSP：小明种苹果续

CSP：小明种苹果续

原题链接

满分代码

#include <iostream>

using namespace std;

int N = 0; // 苹果树数量
int drop[1010] = {0}; // 统计每棵树是否发生掉落
int T = 0;
int D = 0;
int E = 0;

int main()
{
    cin>> N;
    for (int i = 0; i < N; ++i) {
        int m; // 统计的组数
        int num; // 第一个数，树上的苹果数
        int flag = 0; // 是否掉落苹果的标志
        cin >> m;
        cin >> num;
        for (int j = 1; j < m; ++j) {
            int a; // a是一个非正数，下面要用+
            cin>> a;
            if (a <= 0) {
                num += a;
            }
            else {
                if (a != num) { // 发生掉落
                    drop[i] = 1;
                    num = a;
                    flag = 1;
                }
            }
        }
        if (flag == 1) {
            D += 1;
        }
        T += num;
    }
    for (int i = 0; i < N; ++i) {
        if (drop[i] == 1) {
            int pre = (i - 1 + N) % (N);
            int succ = (i + 1 + N) % (N);
            if (drop[pre] == 1 && drop[succ] == 1) {
                E += 1;
            }
        }
    }
    cout << T << " " << D << " " << E << endl;
    return 0;
}