PAT 乙级1012 数字分类 (20分) C++

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

坑点：A2，有可能相加刚好等于0，所以需要设一个flag来标明

#include <iostream>
#include <cmath>
#include <cstdio>

using namespace std;

void A1(int a[],int n);
void A2(int a[],int n);
void A3(int a[],int n);
void A4(int a[],int n);
void A5(int a[],int n);

int main()
{
    int n;
    cin>>n;
    int a[1000];
    for(int i = 0;i<n;i++){
        cin>>a[i];
    }
    A1(a,n);
    A2(a,n);
    A3(a,n);
    A4(a,n);
    A5(a,n);

    return 0;
}

void A1(int a[],int n){
    int sum = 0;
    for(int i = 0;i<n;i++){
        if(a[i]%5==0&&a[i]%2==0){
            sum+=a[i];
        }
    }
    if(sum>0){
        cout<<sum<<" ";
    }
    else{
        cout<<"N"<<" ";
    }
}

void A2(int a[],int n){
    int sum = 0;
    bool flag = false;
    int k = 0;
    for(int i=0;i<n;i++){
        if(a[i]%5==1){
            int t = pow(-1,k)*a[i];
            k++;
            sum+=t;
            flag = true;
        }
    }
    if(flag){
        cout<<sum<<" ";
    }
    else{
        cout<<"N"<<" ";
    }
}

void A3(int a[],int n){
    int count = 0;
    for(int i=0;i<n;i++){
        if(a[i]%5==2){
            count++;
        }
    }
    if(count>0){
        cout<<count<<" ";
    }
    else{
        cout<<"N"<<" ";
    }
}

void A4(int a[],int n){
    double average;
    int count = 0;
    double sum = 0;
    for(int i = 0;i<n;i++){
        if(a[i]%5==3){
            sum+=(double)a[i];
            count++;
        }
    }
    if(count>0){
    average=sum/count;
    printf("%.1lf",average);
    cout<<" ";
    }
    else{
        cout<<"N"<<" ";
    }
}

void A5(int a[],int n){
    int max = 0;
    int flag = 0;
    for(int i=0;i<n;i++){
        if(a[i]%5==4){
            if(a[i]>=a[max]){
                max = i;
                flag = 1;
            }
        }
    }
    if(flag==1){
    cout<<a[max];
    }
    else{
        cout<<"N";
    }
}