PAT乙级1012

PAT1012

题目描述：

给定一系列正整数，请按要求对数字进行分类，并输出以下 5 个数字：

1. $A_1=$ 能被 5 整除的数字中所有偶数的和；
2. $A_2=$ 将被 5 除后余 1 的数字按给出顺序进行交错求和，即计算$n_1-n_2+n_3-n_4\cdots$
3. $A_3=$ 被 5 除后余 2 的数字的个数；
4. $A_4=$ 被 5 除后余 3 的数字的平均数，精确到小数点后 1 位；
5. $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

算法思路与代码实现：

起初做这道题是比较别扭的，本题虽然不难，但是5种分类情况各有独立的处理方法，直接依次对各种情况进行处理虽然最直观、最简单，但是看起来总是不那么优雅美观。本题暂时还没想到足够好的办法，也难于对5种情况统一处理，只能依次计算出结果并输出。

Python实现

numbers = [int(item) for item in input().split()][1:]
kinds = [[item for item in numbers if item % (5 if i else 10) == i] for i in range(5)]
print(sum(kinds[0]) if len(kinds[0]) else 'N', end=' ')
sign = (1, -1)
print(sum(kinds[1][i] * sign[i % 2] for i in range(len(kinds[1]))) if len(kinds[1]) else 'N', end=' ')
print(len(kinds[2]) if len(kinds[2]) else 'N', end=' ')
print(round(sum(kinds[3]) / len(kinds[3]), 1) if len(kinds[3]) else 'N', end=' ')
print(max(kinds[4]) if len(kinds[4]) else 'N')

C++实现

使用C++解题时为了使5种情况分类更清晰，写了5个小函数处理各自的情况，为了使用起来简便，统一这5个函数的参数类型和返回值，以便于利用函数指针数组。当然，不写这5个函数，直接在main()函数里面处理也可以。程序中使用了accumulate()直接求和，对于$A_1$和$A_4$可以简化求和过程。

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

void fun1(vector<int>& v) { cout << accumulate(v.begin(), v.end(), 0) << " "; }
void fun2(vector<int>& v) {
    int interlacedSum = 0, flag = 1;
    for (int i = 0; i < v.size(); i++) {
        interlacedSum += v[i] * flag;
        flag = -flag;
    }
    cout << interlacedSum << " ";
}
void fun3(vector<int>& v) { cout << v.size() << " "; }
void fun4(vector<int>& v) {
    double aver = accumulate(v.begin(), v.end(), 0) / (double)v.size();
    printf("%.1f ", aver);
}
void fun5(vector<int>& v) {
    int m = v[0];
    for (int i = 1; i < v.size(); i++) m = m < v[i] ? v[i] : m;
    cout << m << endl;
}
int main() {
    vector<int> numbers[5];
    int n, m, interlacedSum = 0, sign = 1;
    void (*func[5])(vector<int>&) = {fun1, fun2, fun3, fun4, fun5};
    cin >> n;
    for (int i = 0; i < n; i++) {
        cin >> m;
        if (m % 10 == 0)
            numbers[0].emplace_back(m);
        else if (m % 5 != 0)
            numbers[m % 5].emplace_back(m);
    }
    for (int i = 0; i < 5; i++) {
        if (numbers[i].size())
            func[i](numbers[i]);
        else
            cout << "N" << (i == 4 ? "\n" : " ");
    }
    return 0;
}

在一开始写5个函数的时候，每个函数中都有输出"N"的cout语句，即每个函数都需要判断对应的数字是否存在。后来一想，5个不同的函数却做了同样的一件事情，这是不合理的，因此，将其单独提出来放在5个函数的外面，仅仅让5个函数去处理5个数都存在的情况，这样代码看起来舒服多了。

函数指针也不是必须的，如果不想小题大做的话，完全可以不用函数指针，输入数据之后直接按照题目进行处理。下面是另一种方法实现：

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

int main() {
    vector<int> numbers[5];
    int n, m, sign = 1, a2 = 0, a5 = 0;
    cin >> n;
    for (int i = 0; i < n; i++) {
        cin >> m;
        if (m % 10 == 0)
            numbers[0].emplace_back(m);
        else if (m % 5 != 0)
            numbers[m % 5].emplace_back(m);
    }
    for (int i = 0; i < 5; i++) {
        int length = numbers[i].size();
        for (int j = 0; j < length; j++) {
            if (i == 1) {
                a2 += numbers[i][j] * sign;
                sign = -sign;
            } else if (i == 4)
                a5 = a5 < numbers[i][j] ? numbers[i][j] : a5;
        }
    }
    for (int i = 0; i < 5; i++) {
        if (!numbers[i].size())
            cout << "N" << (i == 4 ? "\n" : " ");
        else {
            if (i == 0)
                cout << accumulate(numbers[0].begin(), numbers[0].end(), 0) << " ";
            else if (i == 1)
                cout << a2 << " ";
            else if (i == 2)
                cout << numbers[2].size() << " ";
            else if (i == 3)
                printf("%.1f ", accumulate(numbers[3].begin(), numbers[3].end(), 0.0) / numbers[3].size());
            else
                cout << a5 << endl;
        }
    }
    return 0;
}