PTA-B1012 数字分类【c/c++】

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

• A1 = 能被 5 整除的数字中所有偶数的和；
• A2 = 将被 5 除后余 1 的数字按给出顺序进行交错求和，即计算 $n_1-n_2+n_3-n_4$⋯；
• 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 <cstdio>
int main()
{
	double a4 = 0, count4 = 0;
	int a1 = 0, a2 = 0, a3 = 0, a5 = 0;
	int count2 = 0, count5 = 0;
	int n;
	int flag = 0;
	while (scanf("%d", &n) != EOF)
		if (!flag) flag = 1;
		else
		{ 
			{
				if (n % 5 == 0 && n % 2 == 0)
				{
					a1 += n;
				}
				if (n % 5 == 1)
				{
					if (count2 % 2 == 0)
					{
						a2 += n;
					}
					else
					{
						a2 -= n;
					}
					count2++;
				}
				if (n % 5 == 2)
				{
					a3++;
				}
				if (n % 5 == 3)
				{
					a4 += n;
					count4 += 1;
				}
				if (n % 5 == 4 && n > a5)
				{
					a5 = n;
					count5++;
				}
			}
	}
	if (a1 == 0) printf("N");
	else printf("%d", a1);
	if (count2 == 0) printf("N");
	else printf(" %d", a2);
	if (a3 == 0) printf("N");
	else printf(" %d", a3);
	if (count4 == 0) printf("N");
	else printf(" %.1f", a4 / count4);
	if (count5 == 0) printf("N");
	else printf(" %d", a5);
}