【PAT_B】1012 数字分类（c/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

思路

利用switch语句，按对应的情况进行计算
设置flag标记判断此类数字是否存在
cout<<fixed<<setprecision(n)<<i;可以输出i保留n位小数后的结果

代码

#include<bits/stdc++.h>

using namespace std;

int main(){
    int n;
    cin>>n;
    int a[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 的数字中最大数字*/
    int j=1;    //用于a[1]交错求和
    int sum=0;  //保存a[3]中数字的个数
    int flag[5];
    fill(a,a+5,0);
    fill(flag,flag+5,0);
    int num;    
    for(int i=0;i<n;i++){
        cin>>num;
        switch(num%5){
            case 0:
                if(num%2==0){
                    a[0]=a[0]+num;
                    flag[0]=1;
                }
                break;
            case 1:
                a[1]=a[1]+j*num;
                j=-j;
                flag[1]=1;
                break;
            case 2:
                a[2]++;
                flag[2]=1;
                break;
            case 3:
                a[3]=a[3]+num;
                sum++;
                flag[3]=1;
                break;
            case 4:
                if(num>a[4]){
                    a[4]=num;
                }
                flag[4]=1;
                break;
        }
    }
    for(int k=0;k<5;k++){
        if(flag[k]==1&&k!=3){
            cout<<a[k];
        }
        else if(flag[k]==1&&k==3){
            cout<<fixed<<setprecision(1)<<a[k]/(double)sum;
        }
        else{
            cout<<"N";
        }
        if(k!=4){
            cout<<" ";
        }
    }
    return 0;
}