PAT 乙级练习 1012

1012. 数字分类

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

代码如下：

#include <iostream>
#include <vector>
#include <iomanip>
using namespace std;

int main() {
    int n, val, tmp;
    vector<int> cnt(5,0);
    vector<bool> option(5,0);
    vector<float> result(5,0);

    cin >> n;
    for(int i=0; i<n; ++i){
        cin >> val;
        tmp = val%5;
        ++cnt[tmp];
        switch(tmp){
        case 0: if(val%2==0) {option[0] = 1; result[0]+=val;}
                break;
        case 1: if(cnt[tmp]%2==1) result[1] += val;
                else result[1] -= val;
                option[1] = 1;
                break;
        case 2: result[2] = cnt[tmp];
                option[2] = 1;
                break;
        case 3: result[3] += val;
                option[3] = 1;
                break;
        case 4: if(result[4] < val) result[4] = val;
                option[4] = 1;
                break;
        }
    }
    result[3] /= cnt[3];

    for(int i=0; i<5; ++i)
        if(i == 0)
            if(option[i] == 0) cout << 'N';
            else            cout << setprecision(0)<<fixed << result[i];
        else if(i == 3)
            if(option[i] == 0) cout << 'N';
            else            cout << ' ' << setprecision(1)<<fixed << result[i];
        else
            if(option[i] == 0) cout << 'N';
            else            cout << ' ' << setprecision(0)<<fixed << result[i];

    return 0;
}