PAT 乙级 1012.数字分类 C++/Java

题目来源

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

分析：

对输入的每个数字num进行mod 5，将得到的值存放在下标为 num % 5的数组中，每次添加一个数

然后分别对每个数组中保存的数字进行处理

C++实现：

#include <iostream>
#include <algorithm>
#include <vector>
using namespace std;

bool myCmp(int a, int b)
{
    return a > b;
}

//数字分类
int main()
{
    // a mod 5最大值为4
    int A1 = ;    //被 5 整除的数字中，偶数的和
    int A2 = ;    //将被 5 除后余 1 的数字按给出顺序进行交错求和，即计算n1 - n2 + n3 - n4...
    int A3 = ;    //被 5 除后余 2 的数字的个数
    double A4 = ; //被 5 除后余 3 的数字的平均数，精确到小数点后 1 位
    int A5 = ;    //被 5 除后余 4 的数字中最大数字 - 排序即可

    int N;
    cin >> N;

    vector<vector<int> > vec(, vector<int>());
    int temp;
    for (int i = ; i < N; ++i)
    {
        cin >> temp;
        vec[temp % ].push_back(temp);
    }

    //求A1;
    int len = vec[].size();
    for (int i = ; i < len; ++i)
    {
        if (vec[][i] %  == )
        {
            A1 += vec[][i];
        }
    }

    //求A2
    len = vec[].size();
    if (len != )
    {
        for (int i = ; i < len; ++i)
        {
            if (i %  == )
            {
                //数组中的第偶数项，取负数 a[1], a[3], a[5]
                // n1 - n2 + n3 - n4
                // 相当于a[0] - a[1] + a[2] - a[3];
                vec[][i] *= -;
            }
            A2 += vec[][i];
        }
    }

    //求A3
    A3 = vec[].size();

    //求A4
    if (vec[].size() != )
    {
        for (int i = ; i < vec[].size(); ++i)
        {
            A4 += vec[][i];
        }
        A4 /= vec[].size();
    }

    //求A5
    if (vec[].size() != )
    {
        sort(vec[].begin(), vec[].end(), myCmp);
        A5 = vec[][];
    }

    for (int i = ; i < ; ++i)
    {
        if (i != )
        {
            cout << " ";    //控制空格输出
        }
        if (i ==  && A1 ==  || i !=  && vec[i].size() == )
        {
            cout << "N";
            continue;
        }
        if (i == )
        {
            cout << A1;
        }
        else if (i == )
        {
            cout << A2;
        }
        else if (i == )
        {
            cout << A3;
        }
        else if (i == )
        {
            printf("%.1f", A4);
        }
        else if (i == )
        {
            cout << A5;
        }
    }
    return ;
}

小结

求A1-A5的时候要判断vector长度是否为0，否则会发生段错误（越界）

求A5可以直接从大到小排序，取第一个数字

输出A4的时候用占位符%.1lf控制输出格式，精确到小数点后一位

循环中判断条件 if (i == && A1 == || i != && vec[i].size() == ) 如果没有 i == && A1 ==  ，若输入0，输出结果就会是0而不是N