PTA(Basic Level) Practice （中文）1012 数字分类 (20 分)

给定一系列正整数，请按要求对数字进行分类，并输出以下 5 个数字：

输入格式：

每个输入包含 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

Code

#include <iostream>
#include <cstdio>
#include <algorithm>
using namespace std;
int main() {
    int a[1005];
    int t;
    int k = 0;
    double A[6] = {0}; 
    bool book[6] = {false}; // 标记是否有某类数
    int flag = 1;
    int n = 0;
    scanf("%d",&t);
    while(t && scanf("%d",&a[k++])) t--;
    for(int i = 0; i < k; i++) {
        switch(a[i]%5) {
        case 0:
            if(a[i]%2 == 0) {
                A[1] += a[i];
                book[1] = true;
            }
            break;
        case 1:
            A[2] += flag*a[i];
            flag = -flag;
            book[2] = true;
            break;
        case 2:
            A[3]++;
            book[3] = true;
            break;
        case 3:
            A[4] += a[i];
            n++;
            book[4] = true;
            break;
        case 4:
            A[5] = max((int)A[5], a[i]);
            book[5] = true;
            break;
        }
    }
    A[4] = 1.0*A[4]/n;
    for(int i = 1; i < 6; i++) {
        switch(i) {
        case 1:
            if(book[1])
                printf("%.f ",A[i]);
            else
                printf("N ");
            break;
        case 2:
            if(book[2])
                printf("%.f ",A[i]);
            else
                printf("N ");
            break;
        case 3:
            if(book[3])
                printf("%.f ",A[i]);
            else
                printf("N ");
            break;
        case 4:
            if(book[4])
                printf("%.1f ",A[i]);
            else 
                printf("N ");
            break;
        case 5:
            if(book[5])
                printf("%.f\n",A[i]);
            else
                printf("N\n");
        }
    }
    return 0;    
}