CCF 小明种苹果

题目

AC代码

#include<bits/stdc++.h>
using namespace std;
int sum=0;
int sum2=0;
int sum3=0;
struct node {
	int num;//数目 
	bool fall;//是否掉落 
};
int main() {
	int n,m;
	cin>>n;
	node*a=new node[n];//储存果树 
	for(int i=0; i<n; i++) {
		a[i].fall=false;//初始化 
		cin>>m;
		int *b=new int[m]; 
		for(int j=0; j<m; j++) {
			cin>>b[j];
		}
		a[i].num=b[0];//最开始的苹果数目 
		for(int j=1; j<m; j++) {
			if(b[j]<=0)//如果因为计算时 掉落 减去 
				a[i].num+=b[j];
			else if(a[i].num>b[j]){//再次计算时更小 
				a[i].fall=true;//说明掉落了 
				a[i].num=b[j];}//数目更新 
		}
		sum+=a[i].num;//计算总苹果数目 
	}
	int x=0;
	for(int i=0; i<n; i++) {
		if(a[i].fall) {
			x++;
			if(x==3) {//连续三个树木均掉落 
				i-=2;//回到前两个 
				x=0;//恢复 
				sum3++;
			}
		} else
			x=0;
		if(i==n-1 and x>=2 and a[0].fall==true) {//讨论第一棵树和最后两棵树存在掉苹果现象 
			sum3++;
		} 
		if(i==n-1 and x>=1 and a[0].fall==true and a[1].fall==true)//同理 
			sum3++;
	}
	for(int i=0; i<n; i++) {//计算掉苹果数目 
		if(a[i].fall)
			sum2++;
	}
	cout<<sum<<" " <<sum2<<" "<<sum3;
}