PAT B1012 数字分类

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

题目代码：需要注意的是最好不要让cout和printf在同一个程序当中出现。

#include<iostream>
#include<vector>
#include<math.h>
using namespace std;

int main() {
  int n;
  cin >> n;
  vector<int> a(n);
  for (int i = 0;i < n;i++) {
    cin >> a[i];
  }
  int A1 = 0, A2 = 0, A3 = 0, A5 = 0, flag2 = 0, j = 0 ;
  double A4 = 0.0, flag4 = 0.0;
  for (int i = 0;i < n;i++) {
    if (a[i] % 5 == 0 && a[i] % 2 == 0) A1 += a[i];
    if (a[i] % 5 == 1) { A2 += (int)(pow(-1, j++)*a[i]); flag2++; }
    if (a[i] % 5 == 2) A3++;
    if (a[i] % 5 == 3) { A4 += a[i]; flag4++; }
    if (a[i] % 5 == 4) if (a[i] > A5) A5 = a[i];
  }
  if (A1 > 0) printf("%d ", A1);
  else printf("N ");

  if (flag2 > 0) printf("%d ", A2);
  else printf("N ");

  if (A3 > 0) printf("%d ", A3);
  else printf("N ");

  if (A4 > 0) printf("%.1f ", A4 / flag4);
  else printf("N ");

  if (A5 > 0) printf("%d", A5);
  else printf("N");

  return 0;
}