CCF認證 201909-2： 小明種蘋果（續）（C++）（80分）

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
const int N=1000;
int a[N][N];
int main()
{	
	int n,i,j,b[N],sum=0,num=0,num1=0,e=0;
	//b[N]記錄是否蘋果掉下來的數組 
	//num1記錄蘋果掉下的數量 
	//sum蘋果總數 
	//num每棵蘋果樹蘋果的數量統計變量 
	//e連續三棵樹都發生蘋果掉下的數量 
	cin>>n;
	memset(a,0,sizeof(a));
	memset(b,0,sizeof(b));
	for(i=0;i<n;i++){
		cin>>a[i][0];
		for(j=1;j<=a[i][0];j++){
			cin>>a[i][j];
		}
	}
	for(i=0;i<n;i++){
		num=a[i][1];
		for(j=2;j<=a[i][0];j++){
			if(a[i][j]<=0){
				num+=a[i][j];
			}else if(a[i][j]>0){
				if(num>a[i][j]){
					b[i]++;
				}
				num=a[i][j];
			}
		}
		sum+=num;
	}
	if(b[n-1]!=0&&b[0]!=0&&b[1]!=0){
		e++;
	}
	if(b[n-2]!=0&&b[n-1]!=0&&b[0]!=0){
		e++;
	}
	for(i=0;i<n;i++){
		if(b[i]!=0){
			num1++;
		}
		if(i!=0&&i!=n-1&&b[i-1]!=0&&b[i]!=0&&b[i+1]!=0){
			e++;
		}
	}
	cout<<sum<<" "<<num1<<" "<<e;
	return 0;
}

只有80分，不知道哪裏出錯，有想法的可以在下面評論