PTA 数字分类（c++）

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 <bits/stdc++.h>
using namespace std;
typedef long long ll;
int main(){
 int n,m;
 cin>>n;
 int a1=0,a2=0,a3=0,a4=0,a5=0;
 int a2_flag=0,a2_sign=1,a4_num=0;
 for(int i=0;i<n;i++){
  cin>>m;
  if(m%5==0&&(m&1)==0){
   a1+=m;
  }
  else if(m%5==1){
   a2_flag=1;
   if(a2_sign){
    a2+=m;
    a2_sign--;
   }
   else{
    a2-=m;
    a2_sign++;
   }
  }
  else if(m%5==2){
   a3++; 
  }
  else if(m%5==3){
   a4+=m;
   a4_num++;
  }
  else if(m%5==4){
   if(m>a5) a5=m;
  }
 }
 if(a1==0)
  cout<<"N ";
 else
  cout<<a1<<" ";
  
 if(a2_flag==0)
  cout<<"N ";
 else
  cout<<a2<<" ";
  
 if(a3==0)
  cout<<"N ";
 else
  cout<<a3<<" ";
  
 if(a4==0)
  cout<<"N ";
 else
  printf("%.1f ",(float)a4/a4_num);
 
 if(a5==0)
  cout<<"N";
 else
  cout<<a5; 
}