PAT 1012 数字分类

1012 数字分类

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

A​1​​ = 能被 5 整除的数字中所有偶数的和；
A​2​​ = 将被 5 除后余 1 的数字按给出顺序进行交错求和，即计算 n​1​​−n​2​​+n​3​​−n​4​​⋯；
A​3​​ = 被 5 除后余 2 的数字的个数；
A​4​​ = 被 5 除后余 3 的数字的平均数，精确到小数点后 1 位；
A​5​​ = 被 5 除后余 4 的数字中最大数字。

输入格式：

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

对给定的 N 个正整数，按题目要求计算 A​1​​~A​5​​ 并在一行中顺序输出。数字间以空格分隔，但行末不得有多余空格。

若其中某一类数字不存在，则在相应位置输出 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 <algorithm>
#include <cstdio>
#include <vector>
#include<string>
#include<cstring>
#include<cmath>
#include<set>
#include<queue>
#include<cmath>
#include<map>
#include<list>
#include<stack>
using namespace std;
const int inf = 0x3f3f3f3f;
typedef long long ll;
const double pi = 3.1415926535;
int arr[1005];
int main() {
    
    int a1 = 0, a2 = 0, a3 = 0, a5 = 0;
    int f1 = 0, f2 = 0, f3 = 0, f4 = 0, f5 = 0;//标志位
    int fuhao = 1;  //交错求和的符号
    double a4 = 0;
    int n;
    cin >> n;
    for(int i=0;i<n;i++)
        cin >> arr[i];
    for (int i = 0; i < n; i++)
    {
        if (arr[i] % 5 == 0 && arr[i] % 2 == 0)
        {
            f1 = 1;
            a1 += arr[i];
        }
        else if (arr[i] % 5 == 1)
        {
            a2 += fuhao * arr[i];
            f2 = 1;
            fuhao *= -1;
        }
        else if (arr[i] % 5 == 2)
        {
            a3++;
            f3 = 1;
        }
        else if (arr[i] % 5 == 3)
        {
            a4 += arr[i];
            f4++;
        }
        else if (arr[i] % 5 == 4)
        {
            a5 = max(a5, arr[i]);
            f5 = 1;
        }
    }
    if (f1)
    {
        cout << a1 << " ";
    }
    else cout << "N ";
    if (f2)
    {
        cout << a2 << " ";
    }
    else cout << "N ";
    if (f3)
    {
        cout << a3 << " ";
    }
    else cout << "N ";
    if (f4)
    {
        printf("%.1lf ", (double)a4 / f4);
    }
    else cout << "N ";
    if (f5)
    {
        cout << a5;
    }
    else cout << "N";
    return 0;
}