PATB1012数字分类

 

给定一系列正整数，请按要求对数字进行分类，并输出以下 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
#include
using namespace std;
int main(){
    int n;
    int number;
    int a1,a3,a4,a5,ans2;
    a1 = a3 = a4 = a5 = ans2 = 0;
    double ans4 = 0.0;
    int count[5] = {0};
    int count4 = 0;
    vector a2;
    scanf("%d",&n);
    for(int i = 0;i < n;i++){
        scanf("%d",&number);
        if((number % 5 == 0) && (number % 2 == 0)){
            a1 += number;
            count[0]++;
        }
        if(number % 5 == 1){
            a2.push_back(number);
            count[1]++;
        }
        if(number % 5 == 2){
            a3++;
            count[2]++;
        }
        if(number % 5 == 3){
            a4 += number;
            count4++;
            count[3]++;
        }
        if(number % 5 == 4){
            if(number > a5){
                a5 = number;
            }
            count[4]++;
        }
    }
    for(int i = 0;i < a2.size();i++){
        if(i % 2 == 0){
            ans2 += a2[i];
        }else{
            ans2 -= a2[i];
        }
    }
    ans4 = (double)a4 / count4;
    if(count[0] == 0) printf("N ");
    else printf("%d ",a1);
    if(count[1] == 0) printf("N ");
    else printf("%d ",ans2);
    if(count[2] == 0) printf("N ");
    else printf("%d ",a3);
    if(count[3] == 0) printf("N ");
    else printf("%0.1f ",ans4);
    if(count[4] == 0) printf("N");
    else printf("%d",a5);
    return 0;
}

注意点：我一开始没有用count数组计算个数，直接判断答案是否为0，导致会有部分错误，因为我的输入数据也可以为0，这样也会输出N，所以一定要用count数组计算个数！！！