PTA 1012數字分類 python滿分解決

題目：

給定一系列正整數，請按要求對數字進行分類，並輸出以下 5 個數字：
A1 = 能被 5 整除的數字中所有偶數的和；
A2 = 將被 5 除後餘 1 的數字按給出順序進行交錯求和，即計算 n1−n​2+n3−n​4⋯；
A​3 = 被 5 除後餘 2 的數字的個數；
A4 = 被 5 除後餘 3 的數字的平均數，精確到小數點後 1 位；
A​5​​ = 被 5 除後餘 4 的數字中最大數字。

輸入格式：

每個輸入包含 1 個測試用例。每個測試用例先給出一個不超過 1000 的正整數 N，隨後給出 N 個不超過 1000 的待分類的正整數。數字間以空格分隔。

輸出格式：

對給定的 N 個正整數，按題目要求計算 A​1​ ~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

注意：

不可以通過A2==0來判斷這一類數字不存在，因爲可能會出現存在數字但相加減爲零的情況。

python實現：

num = map(int , input().split()[1:])
A1,A2,A3,A4,A5 = 0,0,0,0,0
j,p,k = 0,0,0
for i in num:
    if i%10 == 0:
        A1 += i
    if i%5 == 1:
        if j == 0 or j == 2:
            A2 += i
            j = 1
        else:
            A2 -= i
            j = 2
    if i%5 == 2:
        A3 += 1
    if i%5 == 3:
        p = p+1
        k += i
    if i%5 == 4:
        if i > A5:
            A5 = i
if k != 0:
    A4 = '%0.1f'%(k/p)
else:
    A4 = 0
if A1 == 0:
    print('N',end=' ')
else:
    print(A1,end=' ')
if A2 == 0 and j == 0:
    print('N',end=' ')
else:
    print(A2,end=' ')
if A3 == 0:
    print('N',end=' ')
else:
    print(A3,end=' ')
if A4 == 0:
    print('N',end=' ')
else:
    print(A4,end=' ')
if A5 == 0:
    print('N')
else:
    print(A5)