PAT乙级1012 数字分类(Java)

1012 数字分类

分数 20
作者 CHEN, Yue
单位 浙江大学
给定一系列正整数，请按要求对数字进行分类，并输出以下 5 个数字：
A1= 能被 5 整除的数字中所有偶数的和；
A2= 将被 5 除后余 1 的数字按给出顺序进行交错求和;
A3= 被 5 除后余 2 的数字的个数；
A4= 被 5 除后余 3 的数字的平均数，精确到小数点后 1 位；
A5= 被 5 除后余 4 的数字中最大数字。
输入格式：
每个输入包含 1 个测试用例。每个测试用例先给出一个不超过 1000 的正整数 N，随后给出 N 个不超过 1000 的待分类的正整数。数字间以空格分隔。

输出格式：
对给定的 N 个正整数，按题目要求计算 A1~A5，并在一行中顺序输出。数字间以空格分隔，但行末不得有多余空格。
若分类之后某一类不存在数字，则在相应位置输出 N。
输入样例 1：
13 1 2 3 4 5 6 7 8 9 10 20 16 18
输出样例 1：
30 11 2 9.7 9
输入样例 2：
8 1 2 4 5 6 7 9 16
输出样例 2：
N 11 2 N 9

思路：

根据取余结果，用switch-case到对应位置计算。

用vis数组记录该类是否存在数字。

import java.util.Scanner;
public class Main {
    public static void main(String[] args)
    {
        Scanner in=new Scanner(System.in);
        int n=in.nextInt();
        int a1=0,a2=0,a3=0,count=0,a5=0;
        double a4=0;
        boolean[] vis=new boolean[5];
        int tag=1;
        for(int i=0;i<n;i++)
        {
            int a=in.nextInt();
            int t=a%5;
            switch (t)
            {
                case 0:
                    if(a%2==0)
                    {
                        vis[t]=true;
                        a1+=a;
                    }
                    break;
                case 1:
                    vis[t]=true;
                    a2+=tag*a;
                    tag*=-1;//使tag按照1 -1 1 -1...的顺序循环变化
                    break;
                case 2:
                    vis[t]=true;
                    a3++;
                    break;
                case 3:
                    vis[t]=true;
                    a4+=a;
                    count++;
                    break;
                case 4:
                    vis[t]=true;
                    if(a>a5)a5=a;
                    break;
            }
        }
        if(vis[0])System.out.print(a1);
        else System.out.print('N');
        System.out.print(' ');
        if(vis[1])System.out.print(a2);
        else System.out.print('N');
        System.out.print(' ');
        if(vis[2])System.out.print(a3);
        else System.out.print('N');
        System.out.print(' ');
        if(vis[3])System.out.printf("%.1f",a4/count);
        //Java中double类型和float类型都用%f
        else System.out.print('N');
        System.out.print(' ');
        if(vis[4])System.out.print(a5);
        else System.out.print('N');
    }
}