2021全国高校计算机能力挑战赛（初赛）Java试题二

某地大数据防疫平台记录了往来的所有防疫相关信息，包括本地或外地人员、健康码颜色、接种疫苗情况、最近一次核酸结果等

import java.util.ArrayList;
import java.util.Scanner;

public class Main {
    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);
        int N = Integer.parseInt(scanner.nextLine());
        ArrayList inputs = new ArrayList<>();
        for(int i = 0 ; i < N ; i++) {
            inputs.add(scanner.nextLine());
        }
        int yellowNum = 0, redNumber = 0;
        for(int i = 0 ; i < N ; i++) {
            String input = inputs.get(i);
            if(isYellow(input)) yellowNum++;
            else if(isRed(input)) redNumber++;
        }
        System.out.println(yellowNum+" "+redNumber);
    }

    public static boolean isYellow(String input) {
        String[] arr = input.split(" ");
        String localState = arr[0];
        String colorState = arr[1];
        String complete = arr[2];
        String res = arr[3];
        int day = Integer.parseInt(arr[4]);
        if (colorState.equals("Y") || (day < 3 && complete.equals("U") && localState.equals("T"))) {
            return true;
        }
        return false;
    }

    public static boolean isRed(String input) {
        String[] arr = input.split(" ");
        String localState = arr[0];
        String colorState = arr[1];
        String complete = arr[2];
        String res = arr[3];
        int day = Integer.parseInt(arr[4]);
        if (colorState.equals("R") || res.equals("P")) {
            return true;
        }
        return false;
    }
}