csp-小明种苹果（续）

小明种苹果（续）

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

const int N =1010;

int n,m;
int a[N][N];
bool visit[2*N];

int main()
{
    cin >> n;

    for(int i=1;i<=n;i++)
    {
        cin >> a[i][0];
        for(int j=1;j<=a[i][0];j++)
            scanf("%d",&a[i][j]);
    }

    int sum=0,cnt=0;
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=a[i][0];j++)
        {
            if(a[i][j]<=0)
                a[i][1]+=a[i][j];
            else
            {
                if(a[i][j]!=a[i][1])
                {
                    if(!visit[i]) //一棵树掉多次会重复加
                        cnt++;
                    visit[i]=true;
                    a[i][1]=a[i][j];
                }
            }
        }
    }

    for(int i=1;i<=n;i++)
        sum+=a[i][1];

    int cnt1=0;
    for(int i=1;i<=n;i++)  //也可以直接取余，省去对边界条件的判断
    {
        if(i>1 && i<n)
        {
            if(visit[i-1] && visit[i] && visit[i+1])
                cnt1++;
        }
        else if(i==1)
        {
            if(visit[n] && visit[1] && visit[2])
                cnt1++;
        }
        else if(i==n)
        {
            if(visit[n] && visit[n-1] && visit[1])
                cnt1++;
        }
    }

    printf("%d %d %d\n",sum,cnt,cnt1);
    
    return 0;
}
//给出的测试样例
/*
4
4 74 -7 -12 -5
5 73 -8 -6 59 -4
5 76 -5 -10 60 -2
5 80 -6 -15 59 0

5
4 10 0 9 0
4 10 -2 7 0
2 10 0
4 10 -3 5 0
4 10 -1 8 0
*/