1012 数字分类

1012 数字分类

输入格式

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

输出格式

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

输出样例1

N 11 2 N 9

解题思路

首先输入中的第一个数字是数字个数，不在分类数字之中
依次对每种情况进行设置相应的输出
其中使用round函数实现小数点后位数的控制

lst = list(map(int, input().split()))
n = lst[0]
lst.remove(n)

sum= [0]*5
count=[0]*5
for num in lst:
    if num % 10 == 0:
        sum[0] += num
    elif num % 5 == 1:
        sum[1] += (-1) ** count[1] * num
        count[1]+=1
    elif num % 5 == 2:
        sum[2] += 1
    elif num % 5 == 3:
        count[3] += 1
        sum[3] += num
    elif num % 5 == 4:
        sum[4] = max(sum[4], num)

if count[3] == 0:
    sum[3] = "N"
else:
    sum[3] = str(round(sum[3] / count[3], 1))
if count[1] == 0:
    sum[1] = "N"
else:
    sum[1] = str(sum[1])
for i in [0, 2, 4]:
    if sum[i] == 0:
        sum[i] = "N"
    else:
        sum[i] = str(sum[i])

out = " ".join(sum)
print(out)