PAT乙级真题 1012 数字分类 C++实现

题目

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

思路

使用switch语句判断数字类型。需要注意的是：

A2是交替加减，可能结果为0，这时就不能输出’N’。

A3需要计算平均值，并保留1位小数输出：

float average = A[3] * 1.0 / count;  //需先乘一个1.0转化为float，再赋给average
cout << setiosflags(ios::fixed) <<  setprecision(1) << average; //cout二连

算法

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

int main()
{
    vector<int> A(5);
    int n;
    cin >> n;

    int sign = 1;
    bool exist = false; //A2的计算结果可能为0，额外判断是否存在这类数

    A[4] = 0;
    int count = 0;

    int temp;
    for (int i=0; i<n; i++){
        cin >> temp;
        switch (temp%5){
        case 0: if (temp%2==0){
                    A[0] += temp;
                };
                break;
        case 1: A[1] += (temp * sign);
                sign *= -1;
                exist = true;
                break;
        case 2: A[2]++;
                break;
        case 3: A[3] += temp;
                count++;
                break;
        case 4: if (temp>A[4]) {
                    A[4] = temp;
                }
                break;
        }
    }

    if (A[0]==0){
        cout << 'N';
    }
    else{
        cout << A[0];
    }

    for (int i=1; i<5; i++){
        cout << " ";
        if (i==1){
            if (!exist){
                cout << 'N';
            }
            else{
                cout << A[1];
            }
        }
        else if (A[i]==0){
            cout << 'N';
        }
        else if(i==3){
            float average = A[3] * 1.0 / count;
            cout << setiosflags(ios::fixed) <<  setprecision(1) << average;
        }
        else {
            cout << A[i];
        }
    }
    cout << endl;
    return 0;
}