PAT 1012 数字分类 python代码

1012 数字分类

输入样例 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

代码如下

#输入
s= input().split()
n = int(s[0])
num = [int(s[i]) for i in range(1,n+1)]
#分别存放余数为0，1，2，3，4的数字
a1,a2,a3,a4,a5=[],[],[],[],[]
for i in num:
    if i%5==0:
        a1.append(i)
    elif i%5==1:
        a2.append(i)
    elif i%5==2:
        a3.append(i)
    elif i%5==3:
        a4.append(i)
    else:
        a5.append(i)
#aa1,aa2,aa3,aa4,aa5用来存放最终的输出结果
if len(a5)==0:
    aa5 = 'N'
else:
    aa5 = max(a5)

if len(a4)==0:
    aa4 = 'N'
else:
    aa4 = sum(a4)/len(a4)
    aa4 = '%.1f'%aa4

if len(a3)==0:
    aa3 = 'N'
else:
    aa3 = len(a3)

aa2 = 0
for i,j in enumerate(a2):
    if i %2==0:
        aa2+=j
    else:
        aa2-=j
if aa2 == 0:
    aa2 = 'N'
else:
    aa2 = aa2

aa1= 0
for i in a1:
    if i%2==0:
        aa1+=i
if aa1 == 0:
    aa1 = 'N'
else:
    aa1=aa1
#输出
out = str(aa1)+' '+str(aa2)+' '+str(aa3)+' '+str(aa4)+' '+str(aa5)
print(out)

得分