PTA乙级 1012 数字分类 (20)（20 分）

原题链接

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

A
​1
​​ = 能被 5 整除的数字中所有偶数的和；
A
​2
​​ = 将被 5 除后余 1 的数字按给出顺序进行交错求和，即计算 n
​1
​​ −n
​2
​​ +n
​3
​​ −n
​4
​​ ⋯；
A
​3
​​ = 被 5 除后余 2 的数字的个数；
A
​4
​​ = 被 5 除后余 3 的数字的平均数，精确到小数点后 1 位；
A
​5
​​ = 被 5 除后余 4 的数字中最大数字。
输入格式：
每个输入包含 1 个测试用例。每个测试用例先给出一个不超过 1000 的正整数 N，随后给出 N 个不超过 1000 的待分类的正整数。数字间以空格分隔。

输出格式：
对给定的 N 个正整数，按题目要求计算 A
​1
​​ ~A
​5
​​ 并在一行中顺序输出。数字间以空格分隔，但行末不得有多余空格。

若其中某一类数字不存在，则在相应位置输出 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<bits/stdc++.h>
using namespace std;
int main(){
	int a[1001];
	int n; cin>>n;
	for(int i=0; i<n; i++){
		cin>>a[i];
	}
	int A1=0, A2=0, A3=0, A4=0, A5=0;
	int f1=0, f2=0, f3=0, f4=0, f5=0;
	int sign2 = 1, count=0;
	for(int i=0; i<n; i++){
		if(a[i]%5==0 && a[i]%2==0){
			A1 += a[i];
			f1 = 1;
		}
		else if(a[i]%5 == 1){
			A2 += a[i]*sign2;
			sign2 *= -1;
			f2 = 1;
		}
		else if(a[i]%5 == 2){
			A3++;
			f3 = 1;
		}
		else if(a[i]%5 == 3){
			A4 += a[i];
			count++;
			f4 = 1;
		}
		else if(a[i]%5==4){   //这里千万不能用 else, 因为==0时排除了奇数的情况。。。。。 
			if(A5 < a[i]) A5=a[i];
			f5 = 1;
		}
	}
	
	if(f1) cout<<A1;
	else cout<<'N';
	if(f2) cout<<" "<<A2;
	else cout<<" "<<'N';
	if(f3) cout<<" "<<A3;
	else cout<<" "<<'N';
	if(f4) cout<<fixed<<setprecision(1)<<" "<<A4*1.0 / count;
	else cout<<" "<<'N';
	if(f5) cout<<" "<<A5;
	else cout<<" "<<'N';
} 

/*
这道题一定要先理清思路，变量怎么设置，不然后面写起来很乱 
*/