[pat乙]1012 数字分类

1012 数字分类 (20)（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

注意

测试点8 和 可能为0 所以不能当作存在的评判标准

#include <iostream>
#include <algorithm>
#include <iomanip>
using namespace std;

const int MAX = 10001;
int main() {
	int A1 = 0, A2 = 0, A3 = 0, A5[1001], NUM[MAX];
	double A4 = 0;
	int countA2 = 0, countA4 = 0, countA5 = 0;
	int N;
	cin >> N;

	for (int i = 0; i < N; i++) {
		cin >> NUM[i];

		if (!(NUM[i] % 5) && !(NUM[i] % 2))	A1 += NUM[i];

		if (NUM[i] % 5 == 1) {
			countA2++;

			if (!(countA2 % 2))	A2 -= NUM[i];
			else A2 += NUM[i];
		}

		if (NUM[i] % 5 == 2)	A3++;

		if (NUM[i] % 5 == 3)	A4 += NUM[i], countA4++;

		if (NUM[i] % 5 == 4)	A5[countA5] = NUM[i], countA5++;
	}

	A4 /= countA4;

	sort(A5, A5 + countA5, greater<int>());


	if (A1)	cout << A1 << ' ';
	else cout << 'N' << ' ';

	if (countA2)	cout << A2 << ' ';
	else cout << 'N' << ' ';

	if (A3)	cout << A3 << ' ';
	else cout << 'N' << ' ';

	if (A4 && countA4)	cout << fixed << setprecision(1) << A4 << ' ';
	else cout << 'N' << ' ';

	if (A5[0])	cout << A5[0];
	else cout << 'N';

	return 0;
}