PTA乙级1012 数字分类 (20分) -水题

题目原文

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

• A1 = 能被 5 整除的数字中所有偶数的和；
• A2 = 将被 5 除后余 1 的数字按给出顺序进行交错求和，即计算 n1−n2+n3−n4⋯；
• 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

代码如下：

#include<iostream>
#include<cstdio>
#define Wrong 
using namespace std;
int main(void) {
	int N, A;
	long long array[5];
	double a4;
	int Num[5];
	int flag, i;
	scanf("%d", &N);
	for (i = 0; i < 5; i++) {
		array[i] = 0;
		Num[i] = 0;
	}
	flag = 1;
	for (i = 0; i < N; i++) {
		scanf("%d", &A);
		if (A % 5 == 0 && A % 2 == 0) {
			array[0] += A;
			Num[0]++;
		}
		else if (A % 5 == 1) {
			A = A * flag;
			flag = -flag;
			array[1] += A;
			Num[1]++;
		}
		else if (A % 5 == 2) {
			array[2]++;
			Num[2]++;
		}
		else if (A % 5 == 3) {
			array[3] += A;
			Num[3]++;
		}
		else if (A % 5 == 4) {
			if (array[4] < A) {
				array[4] = A;
			}
			Num[4]++;
		}
		else
			;
	}
	if (Num[3] != 0)
		a4 = (double)array[3] / Num[3];
	for (i = 0; i < 4; i++) {
		if (Num[i]==0)
			printf("N ");
		else if (i != 3)
			printf("%lld ", array[i]);
		else if (i == 3)
			printf("%.1f ", a4);
	}
	if (Num[i] == 0)
		printf("N");
	else
		printf("%lld", array[i]);
	return 0;
}