PAT乙级练习题_1012“数字分类”_python解题

原题

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

my answer

input_str = input()
input_number = input_str.split()
A1 = 0
A2 = 0
flag_A2 = False
A2_pre = 1
A3 = 0
A4_times = 0
A4 = 0
A5 = 0
for index in range(int(input_number[0])):
    i = input_number[index+1]
    n = int(i) % 5
    if n == 0:
        if int(i) %  2 == 0:
            A1 += int(i)
    elif n == 1:
        flag_A2 = True
        A2 += A2_pre * int(i)
        A2_pre *= -1
    elif n == 2:
        A3 += 1
    elif n == 3:
        A4 += int(i)
        A4_times += 1
    elif n == 4:
        if A5 < int(i):
            A5 = int(i)

if A1 > 0:
    print(A1, end=' ')
else:
    print('N', end=' ')
if flag_A2:
    print(A2, end=' ')
else:
    print('N', end=' ')
if A3 > 0:
    print(A3, end=' ')
else:
    print('N', end=' ')
if A4_times > 0:
    print("{:.1f}".format(A4/A4_times), end=' ')
else:
    print("N", end=' ')
if A5 > 0:
    print(A5)
else:
    print('N')