PTA 1012 数字分类 python

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

以为挺简单的，但是最后一个样例没通过，不知道哪里出了问题

代码：

x=[int(i) for i in input().split()]
numbers=x[1:]

# 定义 A1, A2, A3, A4, A5
A1 = 0
A2 = 0
A3 = 0
A4 = 0
A5 = 0

cnt=0
cnt2=0

# 遍历给定的数字列表
for number in numbers:
    if number%5==0:
        if number%2==0:
            A1+=number
    elif number%5==1:
        A2+=(-1)**cnt*number
        cnt+=1
    elif number%5==2:
        A3+=1
    elif number%5==3:
        A4+=number
        cnt2+=1
    else:
        A5=max(A5,number)
        
# 输出 A1, A2, A3, A4, A5 的值
print(A1 if A1 else "N", end=" ")
print(A2 if A2 else "N", end=" ")
print(A3 if A3 else "N", end=" ")
print(format(A4/cnt2, ".1f") if cnt2 else "N", end=" ")
print(A5 if A5 else "N")