PAT乙级 1012 数字分类 (python)

1012 数字分类 (20 分)

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

• A₁= 能被 5 整除的数字中所有偶数的和；
• A₂= 将被 5 除后余 1 的数字按给出顺序进行交错求和，即计算 n₁−n₂+n₃−n₄⋯；
• A₃= 被 5 除后余 2 的数字的个数；
• A₄= 被 5 除后余 3的数字的平均数，精确到小数点后 1 位；
• A₅= 被 5 除后余 4 的数字中最大数字。

输入格式：
每个输入包含 1 个测试用例。每个测试用例先给出一个不超过 1000 的正整数 N，随后给出 N 个不超过 1000 的待分类的正整数。数字间以空格分隔。

输出格式：
对给定的 N 个正整数，按题目要求计算 A₁ ~ A₅ 并在一行中顺序输出。数字间以空格分隔，但行末不得有多余空格。
若分类之后某一类不存在数字，则在相应位置输出 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代码：

from functools import reduce
l = list(map(int,input().split()[1:]))
m = {}
A1 = []
A2 = []
A3 = []
a,b,c,d,e = 0,0,0,0,0
f = 1
A4 = []
A5 = []
for i in l:
    if i%10==0:
        d += i
        A1.append(i)
    elif i%5==1:
        A2.append(i)
        e += f * int(i)
        f *= -1
    elif i%5==2:
        A3.append(i)
    elif i%5==3:
        a += i
        b += 1
        A4.append(i)
    else:
        A5.append(i)
m[0] = d if len(A1) else 'N'
m[1] = e if len(A2) else 'N'
m[2] = len(A3) if len(A3) else 'N'
m[3] = round(a/b,1) if len(A4) else 'N'
m[4] = max(A5) if len(A5) else 'N'
print(*m.values())

以上代码部分正确：

全部正确代码（20分）：

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
a = 0
for i in l:
    if i%10 == 0:
        A1 += i
    elif i%5 == 1:
        f += 1
        A2 = A2 + i*(-1)**f
    elif i%5 == 2:
        A3 += 1
    elif i%5 == 3:
        a += 1
        A4 += i
    elif i%5 == 4:
        if A5 <= i:
            A5 = i
if f == 1:
    A2 = 'N'
elif A2 == 0:
    A2 = '0'
if a == 0:
    A4 = 'N'
else:
    A4 = "{:.1f}".format(A4/a)
print(' '.join(list(map(is_o,[A1,A2,A3,A4,A5]))))

或者：

from functools import reduce
s=["sum(i[1])","reduce(lambda acc,q:acc-q[1] if q[0]%2 else acc+q[1],enumerate(i[1]),0)","len(i[1])","sum(i[1])/len(i[1])","max(i[1])"]
group=list(filter(lambda x:x%2==0 or x%5,map(int,input().split()[1:])))
ll=[eval(s[i[0]]) if len(i[1]) else "N"  for i in enumerate([[t for t in group if t%5==i] for i in range(5)])]
print(ll[0],ll[1],ll[2],ll[3] if ll[3]=='N' else "{:.1f}".format(ll[3]),ll[4],sep=" ")