【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

思路：用一个二维数组来记录，A[0][0]到A[4][0]分别记录结果个数，A[0][1]到A[4][1]分别记录结果。对于A3 = 被5除后余2的数字的个数，用A[2][0]同时表示结果个数和结果即可。

#include <cstdio>

int A[5][2];

int main() {
    int N;
    scanf("%d", &N);
    int num;
    for (int i = 0; i < N; i++) {
        scanf("%d", &num);
        if (num % 10 == 0) {
            A[0][0]++;
            A[0][1] += num;
        }
        if (num % 5 == 1) {
            A[1][0]++;
            if (A[1][0] % 2 == 1) A[1][1] += num;
            else A[1][1] -= num;
        }
        if (num % 5 == 2) A[2][0]++;
        if (num % 5 == 3) {
            A[3][0]++;
            A[3][1] += num;
        }
        if (num % 5 == 4) {
            A[4][0]++;
            if (num > A[4][1]) A[4][1] = num;
        }
    }
    if (A[0][0]) { printf("%d ", A[0][1]); } else { printf("N "); }
    if (A[1][0]) { printf("%d ", A[1][1]); } else { printf("N "); }
    if (A[2][0]) { printf("%d ", A[2][0]); } else { printf("N "); }
    if (A[3][0]) { printf("%.1f ", (float) A[3][1] / A[3][0]); } else { printf("N "); }
    if (A[4][0]) { printf("%d\n", A[4][1]); } else { printf("N\n"); }
    return 0;
}