PAT_Basic_1012 数字分类

2022-01-17

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

代码：

#include<iostream>
#include<vector>
#include<iomanip>
using namespace std;
void process(char &s_str,int &num){
    if(s_str!=' '){
        cout<<s_str<<' ';
    }
    else{
        cout<<num<<' ';
    }
}
void floatprocess(char &s_str,float &num){
    if(s_str!=' '){
        cout<<s_str<<' ';
    }
    else{
        cout<<fixed<<setprecision(1)<<num<<' ';
    }
}
int main(){
    int m,n;
    int A1=0,A2=0,A3,A5=0;
    char s_A1=' ',s_A2=' ',s_A3=' ',s_A4=' ',s_A5=' ';
    int count=0;
    float A4=0.0;

    cin>>n;
    vector <int> v;
    for(int i=0;i<n;i++){
        cin>>m;
        v.push_back(m);
    }
    vector <int> v1,v2,v3,v4,v5;
    for(auto it=v.begin();it<v.end();it++){
        if(*it%5==0){
        v1.push_back(*it);
        }
    else if(*it%5==1){
        v2.push_back(*it);
        }
    else if(*it%5==2){
        v3.push_back(*it);
        }
     else if(*it%5==3){
        v4.push_back(*it);
        }
    else if(*it%5==4){
        v5.push_back(*it);
        }

    }
       int count_even=0;
   if(v1.size()==0){
         s_A1='N';
        }
    else{
    for(auto it=v1.begin();it<v1.end();it++){
        if(*it%2==0){
            A1+=*it;
            count_even++;
        } 
    }
    if(count_even==0){
        s_A1='N';
    }

    }
     if(v2.size()==0){
         s_A2='N';
        }
    else{
    for(auto it=v2.begin();it<v2.end();it++){
        if(count%2==0){
            A2+=*it;
        }
        else{
            A2-=*it;
        }
        count++;
    }
    }
     if(v3.size()==0){
         s_A3='N';
        }
    else{
        A3=v3.size();
    }

     if(v4.size()==0){
        s_A4='N';
    }else{
         for(auto it=v4.begin();it<v4.end();it++){
        A4+=*it;
    }
        A4=A4/(v4.size());
    }
     if(v5.size()==0){
         s_A5='N';
    }
    else{
  for(auto it=v5.begin();it<v5.end();it++){
       if(*it>A5){
           A5=*it;}
    }
        }
       process(s_A1,A1);
       process(s_A2,A2);
       process(s_A3,A3);
       floatprocess(s_A4,A4);
        if(s_A5!=' '){
        cout<<s_A5<<endl;
    }
    else{
        cout<<A5<<endl;
    }
    }