PTA 1012 数字分类 python

PTA 1012 数字分类 (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

Python解法

n = input()
num = list(map(int ,n.split()))
A1 = []
A2 = []
A3 = []
A4 = []
A5 = []
# 统计满足条件的值，存入到数组中
for i in num[1:]:
    if i%5 == 0 and i%2 == 0:
        A1.append(i)
    if i%5 == 1:
        A2.append(i)
    if i%5 == 2:
        A3.append(i)
    if i%5 == 3:
        A4.append(i)
    if i%5 == 4:
        A5.append(i)
# print(A1,A2,A3,A4,A5)
s = 0
s1 = 0
s2 = 0
s2_av = 0
# 判断是否有否和条件的值，这里比较麻烦，也可以用数组做
if len(A1) > 0:
    for i in range(0, len(A1)):
        s += A1[i]
    print(s,end = " ")
else:
    print('N',end = " ")
if len(A2) > 0:
    for i in range(0, len(A2)):
        if i % 2 == 1:
            s1 -= A2[i]
            # print(s1)
        else:
            s1 += A2[i]
            # print(s1)
    print(s1,end = " ")
else:
    print('N', end=" ")
if len(A3) > 0:
    print(len(A3),end = " ")
else:
    print('N',end = " ")
if len(A4) > 0:
    for i in range(0, len(A4)):
        s2 += A4[i]
    s2_av = round(s2 / len(A4), 1)
    print(s2_av,end = " ")
else:
    print('N',end = " ")
if len(A5) > 0:
    A5.sort(reverse=True)
    print(A5[0],end = "")
else:
    print('N',end = "")

如有问题，欢迎在评论区提问。