1012 数字分类 (Python)

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

• A1 = 能被 5 整除的数字中所有偶数的和；
• A2 = 将被 5 除后余 1 的数字按给出顺序进行交错求和，即计算 n1−n2+n3−n4⋯；
• A3 = 被 5 除后余 2 的数字的个数；
• A4 = 被 5 除后余 3 的数字的平均数，精确到小数点后 1 位；
• A5 = 被 5 除后余 4 的数字中最大数字。

输入格式：

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

输出格式：

对给定的 N 个正整数，按题目要求计算 A1~A5 并在一行中顺序输出。数字间以空格分隔，但行末不得有多余空格。

若其中某一类数字不存在，则在相应位置输出 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

代码如下：

num = input()
l = num.strip('
').split(' ')
sum_A1 = 0
sum_A2 = 0
sum_A3 = 0
n = 0
num_count = 0
num_A4 = 0
num_A4_count = 0
num_A5_l = []
res = 0
for i in l[1:]:
    i = int(i)
    if i % 5 == 0 and i % 2 == 0:
        sum_A1 += i
    elif i % 5 == 1:
        sum_A2 += (-1)**n * i
        n +=1
    elif i % 5 == 2:
        num_count += 1
    elif i % 5 == 3:
        num_A4 += i
        num_A4_count +=1
    if i % 5 == 4:
        num_A5_l.append(i)
        res = max(num_A5_l)
if sum_A1 == 0:
    sum_A1 = 'N'
if n == 0:
    sum_A2 = 'N'
if num_count == 0:
    num_count = 'N'
if num_A4_count == 0:
    num_4 = 'N'
else:
    num_4 = str(round(num_A4 / num_A4_count,1))
if len(num_A5_l) == 0:
    res = 'N'
print(sum_A1,sum_A2,num_count,num_4,res)