PTA 1012 数字分类(20分) Java实现

题目

给定一系列正整数，请按要求对数字进行分类，并输出以下 5 个数字：

• A​1​​ = 能被 5 整除的数字中所有偶数的和；
• A​2​​ = 将被 5 除后余 1 的数字按给出顺序进行交错求和，即计算 n​1​​−n​2​​+n​3​​−n​4​​⋯；
• A​3​​ = 被 5 除后余 2 的数字的个数；
• A​4​​ = 被 5 除后余 3 的数字的平均数，精确到小数点后 1 位；
• A​5​​ = 被 5 除后余 4 的数字中最大数字。

输入格式：

每个输入包含 1 个测试用例。每个测试用例先给出一个不超过 1000 的正整数 N，随后给出 N 个不超过 1000 的待分类的正整数。数字间以空格分隔。

输出格式：

对给定的 N 个正整数，按题目要求计算 A​1​​~A​5​​ 并在一行中顺序输出。数字间以空格分隔，但行末不得有多余空格。

若其中某一类数字不存在，则在相应位置输出 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

注意

A2要求的交错求和结果可能为0，因此需要添加独立的判断标志检测有无符合条件的输入

AC代码

import java.util.Scanner;

public class Main {
	
	public static void main(String[] args) {
		
		Scanner scanner = new Scanner(System.in);
		int N = scanner.nextInt();
		int[] number = new int[N];
		int[] sum = new int[4];
		
		int flag = 1;
		int a, b, max = 0;
		int count = 0, count1 = 0;
		
		for(int i = 0; i < N; i++) {
			a = scanner.nextInt();
			b = a % 5;
			
			if(b == 0 && a % 2 == 0) {
				sum[0] += a;
			}
			
			if(b == 1) {
				flag = 0;
				sum[1] += Math.pow(-1, count1) * a;
				count1++;
			}
			
			if(b == 2) {
				sum[2]++;
			}
			
			if(b == 3) {
				sum[3] += a;
				count++;
			}
			
			if(b == 4) {
				if(a > max) {
					max = a;
				}
			}
			
		}
		
		for(int i = 0; i < sum.length; i++) {
			
			if(i != 1) {
				if(sum[i] != 0) {
					if(i == 3) {
						System.out.printf("%.1f ", (double)sum[3] / count);
					}else {
						System.out.print(sum[i] + " ");
					}
				}else {
					System.out.print("N ");
				}
			}else {
				if(flag == 0) {
					System.out.print(sum[1] + " ");
				}else {
					System.out.print("N ");
				}
			}
			
		}
		
		if(max != 0) {
			System.out.print(max);
		}else {
			System.out.print("N");
		}
		
	}

}