PTA 乙级 1012 数字分类

给定一系列正整数，请按要求对数字进行分类，并输出以下 5 个数字：
1.A1=能被 5 整除的数字中所有偶数的和；
2.A2= 将被 5 除后余 1 的数字按给出顺序进行交错求和；
3.A3=被 5 除后余 2 的数字的个数；
4.A4= 被 5 除后余 3 的数字的平均数，精确到小数点后 1 位；
5.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<stdio.h>

int main()
{
	int N, *a;

	scanf("%d", &N);
	a = (int*)malloc(sizeof(int)*N);
	for (int i = 0; i < N; i++)scanf("%d", &a[i]);

	int A[5] = { 0,0,0,0,0 }, flag = 1, cnt = 0, judge = 0;

	for (int i = 0; i < N; i++)
	{
		if (a[i] % 5 == 0 && a[i] % 2 == 0)A[0] += a[i];
		else if (a[i] % 5 == 1)
		{
			A[1] += a[i] * flag;
			flag *= -1; judge = 1;
		}
		else if (a[i] % 5 == 2)A[2]++;
		else if (a[i] % 5 == 3)
		{
			A[3] += a[i];
			cnt++;
		}
		else if (a[i] % 5 == 4)
			if (A[4] < a[i])A[4] = a[i];
	}

	for (int i = 0; i < 5; i++)
		if (i == 0)
		{
			if (A[i] == 0)printf("N");
			else printf("%d", A[i]);
		}
		else if (i == 1)
			if (judge == 0)printf(" N");
			else printf(" %d", A[1]);
		else if (i == 3)
		{
			if (A[i] == 0)printf(" N");
			else printf(" %.1f", 1.0*A[i] / cnt);
		}
		else 
		{
			if (A[i] == 0)printf(" N");
			else printf(" %d", A[i]);
		}

	return 0;
}