PAT (Basic Level):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

分析

A2设置flag，一是为了加减交错，二是判断符合A2的数字是否为0。
A3这里设置num是为了之后求平均数

代码

#include<iostream>
using namespace std;
int main(){
  int n,m,flag=0,num=0;
  cin >> n;
  int a[5]={0};//存放A1~A5并初始化 
  for(int i=0;i<n;i++){
    cin >> m;
    if(m%5==0&&m%2==0) a[0]+=m; 
    if(m%5==1){//这里设置flag，一是为了加减交错，二是判断符合A2的数字是否为0。 
      if(flag%2==0) a[1]+=m;
      else a[1]-=m;
      flag++;
    }
    if(m%5==2) a[2]++;
    if(m%5==3){ //这里设置num是为了之后求平均数 
      a[3]+=m;
      num++;
    }
    if(m%5==4&&m>a[4]) a[4]=m; 
  }
  for(int i=0;i<3;i++){
    if(a[i]||(i==1&&flag)) printf("%d ",a[i]);
    else printf("N ");
  }  
  if(a[3]) printf("%.1f ",(float)a[3]/num);
  else printf("N ");
  if(a[4]==0) printf("N");
  else printf("%d",a[4]);
  return 0;
}

测试结果