PAT乙级1012 数字分类 (20 分)

1012 数字分类 (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<iostream>
#include<cstdio>
#include<algorithm> 
#include<cstring>
using namespace std;
void A(int a[],int n){
	int ca[6]={0};
	int i,j,k;
	int sign=1,s1=0,s2=0,s4=0,max=-1;
	double avg=0;
	int flag;
	for(i=0;i<n;i++){
		if(a[i]%5==0&&a[i]%2==0){
			flag=0;
			ca[flag]++;
			s1+=a[i];
		}
		if(a[i]%5==1){
			s2+=sign*a[i];
			sign=-1*sign;
			flag=1;
			ca[flag]++;
		}
		if(a[i]%5==2){
			flag=2;
			ca[flag]++;
		}
		if(a[i]%5==3){
			s4+=a[i];
			flag=3;
			ca[flag]++;
		}
		if(a[i]%5==4){
			flag=4;
			ca[flag]++;
			if(a[i]>max){
				max=a[i];
			}
		}
	}
	avg=(double)s4/ca[3];
		if(ca[0]==0)printf("N ");
		if(ca[0]!=0)printf("%d ",s1);
		if(ca[1]==0)printf("N ");
		if(ca[1]!=0)printf("%d ",s2);
		if(ca[2]==0)printf("N ");
		if(ca[2]!=0)printf("%d ",ca[2]);
		if(ca[3]==0)printf("N ");
		if(ca[3]!=0)printf("%.1f ",avg);
		if(ca[4]==0)printf("N");
		if(ca[4]!=0)printf("%d",max);
}
int main(){
	int N,i;
	cin>>N;
	int a[1001];
	for(i=0;i<N;i++){
		cin>>a[i];
	}
	A(a,N);
	return 0;
}