CSP历年真题201909-2-小明种苹果（续）（Java）

代码

import java.util.Scanner;

public class Main {
    public static void main(String[] args) {
        Scanner in = new Scanner(System.in);
        int n = in.nextInt();
        int t = 0, d = 0, e = 0;
        boolean[] flag = new boolean[n]; // 标记是否发生掉落，true为掉落
        for(int i = 0; i < n; i++){
            int m = in.nextInt();
            int temp = in.nextInt(); // 树上的苹果个数
            for(int j = 2; j < m+1; j++){
                int num = in.nextInt();
                if(num > 0){ // 重新统计苹果树上的苹果树
                    if(temp != num){ // 发生掉落
                        flag[i] = true;
                        temp = num; // 更新树上的苹果个数为num
                    }
                }else{ // num为零或负数，表示进行疏果操作
                    temp += num;
                }
            }
            t += temp;
        }
        in.close();

        for(int i = 0; i < n; i++){
            if(flag[i]){
                d++;
            }
            if(flag[(i-1+n)%n] && flag[(i+n)%n] && flag[(i+1+n)%n]){
                e++;
            }
        }

        System.out.printf("%d %d %d", t, d, e);
    }
}