PAT(Basic Level) 1012 数字分类 (20)

给定一系列正整数，请按要求对数字进行分类。

给定一系列正整数，请按要求对数字进行分类，并输出以下5个数字：

• A1 = 能被5整除的数字中所有偶数的和；
• A2 = 将被5除后余1的数字按给出顺序进行交错求和，即计算n1-n2+n3-n4...；
• A3 = 被5除后余2的数字的个数；
• A4 = 被5除后余3的数字的平均数，精确到小数点后1位；
• A5 = 被5除后余4的数字中最大数字。

输入格式

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

输出格式

对给定的N个正整数，按题目要求计算A1~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

代码实现

Python

n_array = [int(n) for n in input().split()]
A1 = A2 = A3 = A4 = A5 = 0
A2_flag = 1
A2_cnt = 0
A4_cnt = 0

for n in n_array[1:]:
    if n % 5 == 0 and n % 2 == 0:
        A1 += n
    elif n % 5 == 1:
        if A2_flag:
            A2 += n
            A2_flag = 0
        else:
            A2 -= n
            A2_flag = 1
        A2_cnt += 1
    elif n % 5 == 2:
        A3 += 1
    elif n % 5 == 3:
        A4 += n
        A4_cnt += 1
    elif n % 5 == 4:
        if n > A5:
            A5 = n

if A1 == 0:
    A1 = 'N'
if A2_cnt == 0:
    A2 = 'N'
if A3 == 0:
    A3 = 'N'
if A4_cnt > 0:
    A4 /= A4_cnt
    A4 = str('{:.1f}'.format(A4))
else:
    A4 = 'N'
if A5 == 0:
    A5 = 'N'

print(A1, A2, A3, A4, A5)

C语言

#include<stdio.h>

int main() {
    int n;
    int m;
    int A1 = 0;
    int A2 = 0;
    int A3 = 0;
    int A4 = 0;
    int A5 = 0;
    int A2_flag = 1;
    int A2_cnt = 0;
    int A4_cnt = 0;

    scanf("%d", &n);

    for ( int i = 0; i < n; i++ ) {
        scanf("%d", &m);
        if ( m % 5 == 0 && m % 2 == 0 ) {
            A1 += m;
        } else if ( m % 5 == 1 ) {
            if ( A2_flag ) {
                A2 += m;
                A2_flag = 0;
            } else {
                A2 -= m;
                A2_flag = 1;
            }
            A2_cnt++;
        } else if ( m % 5 == 2 ) {
            A3++;
        } else if ( m % 5 == 3 ) {
            A4 += m;
            A4_cnt++;
        } else if ( m % 5 == 4 ) {
            if ( m > A5 ) {
                A5 = m;
            }
        }
    }

    if ( A1 ) {
        printf("%d ", A1);
    } else {
        printf("N ");
    }

    if ( A2_cnt ) {
        printf("%d ", A2);
    } else {
        printf("N ");
    }

    if ( A3 ) {
        printf("%d ", A3);
    } else {
        printf("N ");
    }

    if ( A4_cnt ) {
        printf("%.1f ", ((float)A4) / A4_cnt);
    }else {
        printf("N ");
    }

    if ( A5 ) {
        printf("%d", A5);
    } else {
        printf("N");
    }

    return 0;
}