PAT 1012. 数字分类

给定一系列正整数，请按要求对数字进行分类，并输出以下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

代码如下

#include<iostream>
#include<iomanip>
#include<vector>
#include<math.h>
using namespace std;
int main(){
    int N,n,A1,A2=0,cnt1=0,cnt2=0,cnt3=0,cnt4=0,cnt5=0,max=-1;
    cin>>N;
    double sum=0; 
    for(int i=0;i<N;i++){
       cin>>n;
       if(n%2==0&&n%5==0){
          A1+=n; cnt1++;
       }else if(n%5==1)
          A2+=pow(-1,cnt2++)*n;
       else if(n%5==2) 
          cnt3++;
       else if(n%5==3){
          cnt4++; sum+=n;
       }else if(n%5==4&&n>max){
          max=n; cnt5++;
       }
    }
    if(cnt1==0) cout<<"N "; else cout<<A1<<" ";
    if(cnt2==0) cout<<"N "; else cout<<A2<<" ";
    if(cnt3==0) cout<<"N "; else cout<<cnt3<<" ";
    if(cnt4==0) cout<<"N "; else cout<<setiosflags(ios::fixed)<<setprecision(1)<<sum/cnt4<<" ";
    if(cnt5==0) cout<<"N"; else cout<<max;
    return 0;
}