PAT 1012. 数字分类

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

• #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;
  }