【PAT乙级】1012 数字分类

问题描述

输入格式
每个输入包含 1 个测试用例。每个测试用例先给出一个不超过 1000 的正整数 N，随后给出 N 个不超过 1000 的待分类的正整数。数字间以空格分隔。

输出格式
对给定的 N 个正整数，按题目要求计算 A​1~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

C++代码

#include <bits/stdc++.h>
using namespace std;
int main() {
    int n;
    cin >> n;
    int a1 = 0, a2 = 0, a3 = 0, a5 = 0, b = -1, a4n = 0;
    bool flag[5] = {0};
    double a4sum = 0;
    while(n--) {
        int t;
        cin >> t;
        if(t%5==0 && t%2==0) {a1 += t; flag[0] = 1;}
        if(t%5 == 1) {
            b==-1 ? b=1 : b=-1;
            a2 += t*b;
            flag[1] = 1;
        }
        if(t%5 == 2) {a3++; flag[2] = 1;}
        if(t%5 == 3) {
            a4sum += t;
            a4n++;
            flag[3] = 1;
        }
        if(t%5==4 && t>a5) {a5 = t; flag[4] = 1;}
    }
    if(flag[0]) {
        printf("%d ", a1);
    }else printf("N ");
    if(flag[1]) {
        printf("%d ", a2);
    }else printf("N ");
    if(flag[2]) {
        printf("%d ", a3);
    }else printf("N ");
    if(flag[3]) {
        printf("%.1lf ", a4sum*1.0/a4n);
    }else printf("N ");
    if(flag[4]) {
        printf("%d", a5);
    }else printf("N");
    return 0;
}