《算法笔记》3.1小节——入门模拟-＞简单模拟 问题 G: 数字分类 (20)

问题 G: 数字分类 (20)

时间限制 : 1.000 sec 内存限制 : 32 MB

题目描述

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

A₁ = 能被5整除的数字中所有偶数的和；

A₂ = 将被5除后余1的数字按给出顺序进行交错求和，即计算n₁-n₂+n₃-n₄…；

A₃ = 被5除后余2的数字的个数；

A₄ = 被5除后余3的数字的平均数，精确到小数点后1位；

A₅ = 被5除后余4的数字中最大数字。

输入

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

输出

对给定的N个正整数，按题目要求计算A₁~ A₅并在一行中顺序输出。数字间以空格分隔，但行末不得有多余空格。

若其中某一类数字不存在，则在相应位置输出“N”。

样例输入

13 1 2 3 4 5 6 7 8 9 10 20 16 18
8 1 2 4 5 6 7 9 16

样例输出

30 11 2 9.7 9
N 11 2 N 9

此题与PAT乙级1012相同

#include <bits/stdc++.h>

using namespace std;
using gg = long long;

int main() {
    int N;
    cin >> N;
    int a, flag = 1;
    int A1 = 0, A2 = 0, A3 = 0, A4_count = 0, A5 = -1, A2_count = 0;
    double A4 = 0;
    for (int i = 0; i < N; ++i) {
        cin >> a;
        if (a % 5 == 0 && a % 2 == 0) {
            A1 += a;
        } else if (a % 5 == 1) {
            A2 += flag * a;
            flag = -flag;
            A2_count++;
        } else if (a % 5 == 2) {
            A3++;
        } else if (a % 5 == 3) {
            A4_count++;
            A4 += a;
        } else if (a % 5 == 4) {
            A5 = max(A5, a);
        }
    }
    if (A1 == 0) {
        printf("N ");
    } else {
        printf("%d ", A1);
    }
    if (A2_count == 0) {
        printf("N ");
    } else {
        printf("%d ", A2);
    }if (A3 == 0) {
        printf("N ");
    } else {
        printf("%d ", A3);
    }if (A4_count == 0) {
        printf("N ");
    } else {
        printf("%.1f ", A4/A4_count);
    }if (A5 == -1) {
        printf("N");
    } else {
        printf("%d", A5);
    }
    return 0;
}