PAT (Basic Level) Practice （中文）1012 数字分类 python

出处：
https://pintia.cn/problem-sets/994805260223102976/problems/994805311146147840
这个题比较容易，不过一些小细节要注意。
代码：

m = list(map(int,input().split()))
n = m[0]
res = m[1:]
A1,A2,A3,A4,A5 = 0,0,0,0,0
flag = 1
sum1 = 0
sum2 = 0
for i in range(n):
    if res[i] % 5 == 0 and res[i] % 2 == 0:
        A1 += res[i]
    if res[i] % 5 == 1:
        sum1 = 1
        A2 += flag * res[i]
        flag = -flag
    if res[i] % 5 == 2:
        A3 += 1
    if res[i] % 5 == 3:
        A4 += res[i]
        sum2 += 1
    if res[i] % 5 == 4:
        if res[i] > A5:
            A5 = res[i]
if A1 == 0:
    A1 = 'N'
if sum1 == 0:
    A2 = 'N'
if A3 == 0:
    A3 = 'N'
if A4 == 0:
    A4 = 'N'
else:
    A4 = '%.1f'%(A4 / sum2)
if A5 == 0:
    A5 = 'N'
print(A1,A2,A3,A4,A5)

第一次写的错误代码：

n = list(map(int,input().split()))
res = n[1:]
m = len(res)
A1 = 0
for i in range(m):
    if res[i] % 5 == 0 and res[i] % 2 == 0:
        A1 += res[i]
if A1 == 0:
    A1 = 'N'
A2 = 0
flag = 1
for i in range(m):
    if res[i] % 5 == 1:
        A2 += flag * res[i]
        flag = -flag
if A2 == 0:
    A2 = 'N'
A3 = 0
for i in range(m):
    if res[i] % 5 == 2:
        A3 += 1
if A3 == 0:
    A3 = 'N'
A4 = 0
sum = 0
for i in range(m):
    if res[i] % 5 == 3:
        A4 += res[i]
        sum += 1
if A4 == 0:
    A1 = 'N'
else:
    A4 = float(A4) / sum
    A4 = ('%.1f' % A4)
A5 = 0
for i in range(m):
    if res[i] % 5 == 4:
        if res[i] > A5:
            A5 = res[i]
if A5 == 0:
    A5 = 'N'
print(A1,end = ' ')
print(A2,end = ' ')
print(A3,end = ' ')
print(A4,end = ' ')
print(A5)

开始的时候我是把每种情况分开写的，这样会造成很多此没有必要的循环。
而且其中还有很多问题。
比如说在要求里，“若其中某一类数字不存在，则在相应位置输出 N。” 其他可以直接这样判断：

if A4 == 0:
    A1 = 'N'

但是其中A2和A4要做特殊处理。
改过以后的代码是这样的：
错误代码：

m = list(map(int,input().split()))
n = m[0]
res = m[1:]
A1,A2,A3,A4,A5 = 0,0,0,0,0
flag = 1
sum1 = 0
sum2 = 0
for i in range(n):
    if res[i] % 5 == 0 and res[i] % 2 == 0:
        A1 += res[i]
    if res[i] % 5 == 1:
        sum1 = 1
        A2 += flag * res[i]
        flag = -flag
    if res[i] % 5 == 2:
        A3 += 1
    if res[i] % 5 == 3:
        A4 += res[i]
        sum2 += 1
    if res[i] % 5 == 4:
        if res[i] > A5:
            A5 = res[i]
if A1 == 0:
    A1 = 'N'
if sum1 == 0:
    A2 = 'N'
if A3 == 0:
    A3 = 'N'
if A4 == 0:
    A1 = 'N'
else:
    A4 = '%.1f'%(A4 / sum2)
if A5 == 0:
    A5 = 'N'
print(A1,A2,A3,A4,A5)

看了半天不知道哪里错了，后来发现：

if A4 == 0:
    A1 = 'N'

把A4写成A1了，大无语。