CSP 201909-2 小明种苹果（续）

一、问题链接

CSP 201909-2 小明种苹果（续）

二、思路

模拟
用苹果树的编号对总数取余，模拟树围成一圈

三、参考代码

#include <bits/stdc++.h>
using namespace std;

struct tree {
	int t;//苹果总数
	int d;//d=0为苹果未掉落
	tree(int a,int b):t(a),d(b) {}
};
int main() {
	ios::sync_with_stdio(false);
	cin.tie(0);
	int ni,mi,ai;
	cin>>ni;
	vector<tree> v;
	for(int j=0; j<ni; j++) {
		cin>>mi>>ai;
		int count=ai,flag=0;//记录苹果树,是否掉落
		for(int i=1; i<mi; i++) {
			cin>>ai;
			if(ai<=0) count+=ai;//蔬果
			else {//重新统计
				if(count!=ai) { //掉落
					count=ai;//重新算数
					flag=1;
				}
			}
		}
		v.push_back(tree(count,flag));
	}
	int t=0,d=0,e=0;
	for(auto &i:v) {
		t+=i.t;//苹果总数
		if(i.d==1) {
			d+=1;//掉落苹果的树的数量
		}
	}
	for(int i=0; i<ni; i++) {//用苹果树的编号取余的方法，模拟树围成一圈 
		if(v[i].d==1&&v[(i+1)%ni].d==1&&v[(i+2)%ni].d==1) e++;
	}
	cout<<t<<" "<<d<<" "<<e;
	return 0;
}