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

思路：我的方法比较直接，将所有的数字存在vector中，写上5个函数one ~ five分别用于计算A1~A5的值并返回，最后根据每个函数求得的结果来确定输出就行了。

代码：

#include<bits/stdtr1c++.h>
using namespace std;
int one(vector<int> v1) {
	int sum1 = 0;
	for (int i = 0; i < int(v1.size()); i++) {
		if (v1[i] % 5 == 0 && v1[i] % 2 == 0)
			sum1 += v1[i];
	}
	return sum1;
}
int two(vector<int> v2) {
	int sum2 = 0, flag = 1, cnt = 0;
	for (int i = 0; i < int(v2.size()); i++) {
		if (v2[i] % 5 == 1) {
			cnt++;
			sum2 += v2[i] * flag;
			flag *= -1;
		}
	}
	if (cnt != 0)
		return sum2;
	else
		return -1;
}
int three(vector<int> v3) {
	int sum3 = 0;
	for (int i = 0; i < int(v3.size()); i++) {
		if (v3[i] % 5 == 2) sum3++;
	}
	return sum3;
}
double four(vector<int> v4) {
	double sum4 = 0, cnt = 0;
	for (int i = 0; i < int(v4.size()); i++) {
		if (v4[i] % 5 == 3) {
			sum4 += v4[i];
			cnt++;
		}
	}
	if (cnt == 0) return 0;
	else return sum4 / cnt;
}
int five(vector<int> v5) {
	int max_num = 0;
	for (int i = 0; i < int(v5.size()); i++) {
		if (v5[i] % 5 == 4) {
			if (v5[i] > max_num)
				max_num = v5[i];
		}
	}
	return max_num;
}
int main() {
	int n, num;
	vector<int> v;
	cin >> n;
	for (int i = 0; i < n; i++) {
		cin >> num;
		v.emplace_back(num);
	}
	if (one(v) != 0) cout << one(v);
	else cout << "N";
	if (two(v) != -1) cout << " " << two(v);
	else cout << " N";
	if (three(v) != 0) cout << " " << three(v);
	else cout << " N";
	if (four(v) != 0.0) printf(" %.1f", four(v));
	else cout << " N";
	if (five(v) != 0) cout << " " << five(v);
	else cout << " N";
	return 0;
}