PAT:数字分类

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

输入例子:
13 1 2 3 4 5 6 7 8 9 10 20 16 18

输出例子:
30 11 2 9.7 9

#include <iostream>
#include <string.h>
#include <cstdio>
int main () {
  int n;
  std::cin>>n;

  int str[n];
  for(int i = 0; i < n; i++)
      std::cin>>str[i];

  int a1 = 0,a2 = 0,a3=0,a5 = -1,t = 0,s[n],s1 = -1,q = 1;
  double a4=0.0;
  memset(s,0,sizeof(s));

  for(int i = 0; i < n; i++)
  {
      if(str[i] % 5 == 0 && str[i] %2 == 0) a1 += str[i]; 
      if(str[i] % 5 == 1) { a2 += str[i] * q; q *= -1; }
      if(str[i] % 5 == 2) a3++;
      if(str[i] % 5 == 3) { a4 += (double)str[i]; t++; }
      if(str[i] % 5 == 4) s[s1++] = str[i]; 
  }

  if(t) a4 = a4 / t * 1.0;
  for(int i = 0; i < n; i++) if(s[i] > a5) a5 = s[i];

  if(a1 == 0) std::cout<<"N ";
  else std::cout<<a1<<" ";

  if(a2 == 0) std::cout<<"N ";
  else std::cout<<a2<<" ";

  if(a3 == 0) std::cout<<"N ";
  else std::cout<<a3<<" ";

  if(a4 == 0) std::cout<<"N ";
  else printf("%.1lf ",a4);

  if(a5 == 0) std::cout<<"N";
  else std::cout<<a5;
  return 0;
}