PAT1012 数字分类 (20 分)

题目描述

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

C++解法

解法1

#include<cstdio>
using namespace std;
int main(){
    int ans[5]={0};
    int count[5]={0};
    int n,temp;
    scanf("%d",&n);
    for(int i=0;i<n;i++){
        scanf("%d",&temp);
        if(temp%5==0){
            if(temp%2==0){
                ans[0]+=temp;
                count[0]++;
            }
        }else if(temp%5==1){
            if(count[1]%2==0){
                ans[1]+=temp;
            }else{
                ans[1]-=temp;
            }
            count[1]++;
        }else if(temp%5==2){
            count[2]++;
        }else if(temp%5==3){
            ans[3]+=temp;
            count[3]++;
        }else{
            if(temp>ans[4]){
                ans[4]=temp;
            }
            count[4]++;
        }
    }
    if(count[0]==0) printf("N ");
    else printf("%d ",ans[0]);
    if(count[1]==0) printf("N ");
    else printf("%d ",ans[1]);
    if(count[2]==0) printf("N ");
    else printf("%d ",count[2]);
    if(count[3]==0) printf("N ");
    else printf("%.1f ",(double)ans[3]/count[3]);
    if(count[4]==0) printf("N");
    else printf("%d",ans[4]);
    return 0;
}

解法2

#include <iostream>
#include <vector>
using namespace std;
int main() {
    int n, num, A1 = 0, A2 = 0, A5 = 0;
    double A4 = 0.0;
    cin >> n;
    vector<int> v[5];
    for (int i = 0; i < n; i++) {
        cin >> num;
        v[num%5].push_back(num);
    }
    for (int i = 0; i < 5; i++) {
        for (int j = 0; j < v[i].size(); j++) {
            if (i == 0 && v[i][j] % 2 == 0) A1 += v[i][j];
            if (i == 1 && j % 2 == 0) A2 += v[i][j];
            if (i == 1 && j % 2 == 1) A2 -= v[i][j];
            if (i == 3) A4 += v[i][j];
            if (i == 4 && v[i][j] > A5) A5 = v[i][j];
        }
    }
    for (int i = 0; i < 5; i++) {
        if (i != 0) printf(" ");
        if (i == 0 && A1 == 0 || i != 0 && v[i].size() == 0) {
            printf("N"); continue;
        }
        if (i == 0) printf("%d", A1);
        if (i == 1) printf("%d", A2);
        if (i == 2) printf("%d", v[2].size());
        if (i == 3) printf("%.1f", A4 / v[3].size());
        if (i == 4) printf("%d", A5);
    }
    return 0;
}

python解法

l = list(map(int,input().split()))[1:]

def is_o(Ai):
    if isinstance(Ai, str):
        return Ai
    elif Ai == 0:
        return 'N'
    else:
        return str(Ai)

A1, A2, A3, A4, A5 = 0, 0, 0, 0, 0

f_1 = 1
f_2 = 0
for i in l:
    if i%10 == 0:
        A1 += i
    elif i%5 == 1:
        f_1 += 1
        A2 = A2 + i*(-1)**f_1
    elif i%5 == 2:
        A3 += 1
    elif i%5 == 3:
        f_2 += 1
        A4 += i
    elif i%5 == 4:
        if A5 <= i:
            A5 = i

if f_1 == 1:
    A2 = 'N'
elif A2 == 0:
    A2 = '0'

if f_2 == 0:
    A4 = 'N'
else:
    A4 = "{:.1f}".format(A4/f_2)

print(' '.join(list(map(is_o,[A1,A2,A3,A4,A5]))))

综合比较还是觉得第一种方法最好用