PAT乙级 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。

二.代码

#include<bits/stdc++.h>
using namespace std;
int main()
{
    int a[5]={0},ex=0,num=0,f[5]={0},t,x;
    cin>>t;
    while(t--)
    {
        cin>>x;
        switch(x%5)
        {
            case 0:
                if(x%2==0)
                {
                    a[0]+=x;
                    f[0]=1;
                }
                break;
            case 1:
                a[1]+=x*pow(-1,ex);
                f[1]=1;
                ex++;
                break;
            case 2:
                a[2]++;
                f[2]=1;
                break;
            case 3:
                a[3]+=x;
                num++;
                f[3]=1;
                break;
            case 4:
                if(x>a[4])
                {
                    a[4]=x;
                    f[4]=1;
                }
                break;
        }
    }
    if(f[0]==1) cout<<a[0]<<" ";
    else        cout<<"N ";
    if(f[1]==1) cout<<a[1]<<" ";
    else        cout<<"N ";
    if(f[2]==1) cout<<a[2]<<" ";
    else        cout<<"N ";
    if(f[3]==1) cout<<fixed<<setprecision(1)<<(double)a[3]/num<<" ";
    else        cout<<"N ";
    if(f[4]==1) cout<<a[4]<<endl;
    else        cout<<"N"<<endl;
}